第八星系人气爱 发表于 2024-4-2 23:39:10

深度学习|安装图神经网络必备神器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




安装成功。


本篇文章来源于微信公众号:第八星系人造大气理论爱好者



继续宠爱 发表于 2024-4-27 19:27:58

我也在研究这个问题,希望我们可以交流一下经验。
页: [1]
查看完整版本: 深度学习|安装图神经网络必备神器PyTorch Geometric