第一步:查看已安装的pytorch版本和cuda版本
1.1、进入Anaconda Prompt激活环境
1.2、输入conda list查看此环境下已安装的第三方库版本列表
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
2.2、进入后找到对应版本的四个轮子文件:
torch_scatter、torch_sparse、torch_cluster、torch_spline_conv
注意:cp38代表python3.8版本,win代表windows系统
2.3、四个轮子文件各下载一个,前面的小版本无所谓,点击链接即可下载
2.4、将下载好的安装包剪切到C盘用户文件夹(你电脑现在运行的账户)下面
只有把下载好的轮子文件安装包放在这个路径,才能在anaconda里面直接使用pip安装这四个轮子文件
第三步:安装已经下载的轮子文件安装包
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
第五步:安装成功-运行试试