赞
踩
** (注意)请首先确定自己的pytorch和cuda对应的版本号**
在pycharm中输入以下代码:
import torch
print (torch.__version__)
得到对应的版本号:
然后我们进入这个网站中下载对应的安装包:https://pytorch-geometric.com/whl/
接着找到与之对应的torch版本和cuda版本:
点击链接进入之后(接下来这一步非常关键):
在终端(注意自己的虚拟环境)或者在pycharm中输入以下代码:
pip debug --verbose # 此命令可以查看自己平台支持的版本
看到上图中我的是cp38-cp38-win_amd64,所以我们点进去链接
找到这四个对应的安装包进行下载(注意都是cp38-cp38-win_amd64)
下载完成之后,在终端进入这个四个安装包所在的目录,直接执行
pip install torch_cluster-1.6.1+pt20cu118-cp38-cp38-win_amd64.whl torch_scatter-2.1.1+pt20cu118-cp38-cp38-win_amd64.whl torch_sparse-0.6.17+pt20cu118-cp38-cp38-win_amd64.whl torch_spline_conv-1.2.2+pt20cu118-cp38-cp38-win_amd64.whl
你们需要找到自己对应的安装包进行安装(以下是我安装的截图):
在成功安装第一步中的四个包之后,直接执行在终端执行代码:
pip install torch_geometric
然后就安装成功啦!哈哈哈哈哈哈哈哈!!!!!!!!!!!!!!!!!!!!!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。