pytorch下载地址
确定cuda版本,找对对应的torch、torchvision下载:链接: pytorch
下载完成之后,放到指定路径下(使用虚拟环境,需要在虚拟环境下安装),并在该路径下执行安装命令:
pip install torch-1.5.0-cp38-cp38-linux_x86_64.whl
geometric下载地址
确定需要下载的geometric版本,这里以 torch-geometric 1.5.0 为例,使用清华源快速安装geometric:命令如下:
pip install torch-geometric==1.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
【只需改动其中的版本号即可,附上geometric清华源链接 https://pypi.tuna.tsinghua.edu.cn/simple/torch-geometric/ 】
除此以外,需要下载geometric的依赖包troch_sparse、torch_scatter、torch_cluster以及torch_spline_conv,从 geometric 这个连接中找到对应torch+geometric版本点击进去,分别下载上述四个包(roch_sparse、torch_scatter、torch_cluster以及torch_spline_conv)。
安装命令如下(举例而已):
pip install torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
pip install torch_scatter-2.0.8-cp38-cp38-linux_x86_64.whl
pip install torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl
pip install torch_sparse-0.6.12-cp38-cp38-linux_x86_64.whl
【下载所使用的系统(linux or win)、python对应的版本,若使用的是python 3.8,则找含cp38关键字的包下载。若一个包有好几个版本,选最新的下载即可】