深度学习|安装图神经网络必备神器PyTorch Geometric
本帖最后由 第八星系人气爱 于 2024-4-2 23:40 编辑图神经网络模型对气象是一种非常重要的研究手段。天气、气候的预报本身可以抽象为时空预测,而图神经网络恰恰是“空间”方面的重要基础。
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
安装成功。
本篇文章来源于微信公众号:第八星系人造大气理论爱好者
我也在研究这个问题,希望我们可以交流一下经验。
页:
[1]