一、安装torch
方式一:在pytorch官网查找命令
方式二:下载对应版本的whl到本地,进入whl所在文件夹使用pip安装
※ 注意torch1.8及以后才可以使用torch_geometric
e.g.:
pip install torch-版本信息.whl
二、安装四个相关库
从pytorch-geometric的whl网站中找到对应torch和cuda版本的4个whl,下载完毕后手动pip安装。
注意看好环境的python版本信息,cp3x
对应python3.x
e.g.:(自己安装时记得替换相关版本号)
pip install torch_cluster-1.5.9-cp38-cp38-win_amd64.whl
pip install torch_scatter-2.0.9-cp38-cp38-win_amd64.whl
pip install torch_sparse-0.6.12-cp38-cp38-win_amd64.whl
pip install torch_spline_conv-1.2.1-cp38-cp38-win_amd64.whl
三、安装torch-geometric
pip install torch-geometric