深度学习|安装图神经网络必备神器PyTorch Geometric

文摘   2024-07-17 09:00   河北  

深度学习

安装图神经网络必备神器

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

安装成功。


END


后台私信:第八星系

群内每日更新分享数据

进群回复 第八星系

本文编辑 | Eva


第八星系人造大气理论爱好者
记录与交流python、matlab等科研工具。记录与交流大气科学的学科知识
 最新文章