深度学习
安装图神经网络必备神器
PyTorch Geometric
作者:第八星系-李智
邮箱:lizhi258147369@163.com
前言
图神经网络模型对气象是一种非常重要的研究手段。天气、气候的预报本身可以抽象为时空预测,而图神经网络恰恰是“空间”方面的重要基础。
1.PyG的github地址
https://github.com/pyg-team/pytorch_geometric
2.切忌直接pip安装,我们需要找到下图这个位置,再点击here。
2. 这里回顾
https://mp.weixin.qq.com/s?__biz=MzkyNjUwMzQ5Mg==&mid=2247486321&idx=1&sn=9105f963baa43a979e12a1d13788e59f&chksm=c2370c5af540854caa673fbb9a54953430bc081c5c2fbdac079df2540e5a98e47d13c83ceb94&mpshare=1&scene=23&srcid=0330ipnWCbyQTcZbc2NOCNG9&sharer_shareinfo=0999850c4089194654d5893dd005ad30&sharer_shareinfo_first=0999850c4089194654d5893dd005ad30#rd
我们安装的torch、cuda与python的版本。
2. 安装这四个文件。
这里注意,因为我的python有多个环境,现在这个是base,而我要激活别的环境,在那个环境上安装pyg.
输入pip install torch-geometric
安装成功。
后台私信:第八星系
群内每日更新分享数据
进群回复 第八星系
本文编辑 | Eva