第一步:查看已安装的pytorch版本和cuda版本

1.1、进入Anaconda Prompt激活环境

d75924a1228d4c308e34c2b9cc02a377.png

1.2、输入conda list查看此环境下已安装的第三方库版本列表

fb587f7bed334e3996773141a9da30e6.png

bfd6eb4bc4f34c29b5f1c88028309018.png

pytorch版本为2.0.1、cuda版本为11.7、python版本为3.8

 

第二步:进入网址寻找轮子文件进行下载

网址:https://pytorch-geometric.com/whl/

2.1、ctrl+f查找:cu117

找到对应pytorch和cuda版本的链接点进去,cu117就代表cuda版本是11.7,而我pytorch版本是2.0.1

b729ec4ab8044832a8523177ebca7f3b.png

2.2、进入后找到对应版本的四个轮子文件:

torch_scatter、torch_sparse、torch_cluster、torch_spline_conv

注意:cp38代表python3.8版本,win代表windows系统

ccc071c6e54740fcb8452b16dc846bba.png

2.3、四个轮子文件各下载一个,前面的小版本无所谓,点击链接即可下载

2.4、将下载好的安装包剪切到C盘用户文件夹(你电脑现在运行的账户)下面

只有把下载好的轮子文件安装包放在这个路径,才能在anaconda里面直接使用pip安装这四个轮子文件

e24b002170244033b2c4ba6729a46a51.png

 

第三步:安装已经下载的轮子文件安装包

3.1、依次输入四句命令:

pip install torch_scatter-2.1.2+pt20cu117-cp38-cp38-win_amd64.whl
pip install torch_sparse-0.6.18+pt20cu117-cp38-cp38-win_amd64.whl
pip install torch_cluster-1.6.3+pt20cu117-cp38-cp38-win_amd64.whl
pip install torch_spline_conv-1.2.2+pt20cu117-cp38-cp38-win_amd64.whl

完成轮子文件的安装

 

第四步:直接pip安装geometric

4.1、输入命令:


pip install torch-geometric

或者加上清华大学镜像源:
pip install torch-geometric -i https://pypi.tuna.tsinghua.edu.cn/simple

 

第五步:安装成功-运行试试

ad397b804ec445ea949a387e15eb564e.png

c7382f73baa946359aacc15023143da2.png

 

 

 

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐