- 到官网中 https://pytorch-geometric.com/whl/torch-1.6.0.html
下载对应的版本 cu为cuda版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
https://pytorch-geometric.com/whl/torch-1.6.0%2Bcu102.html
我这里是10.2, cp为python 版本。(可以用win浏览器下载传进linux)
linux 下载示例
wget https://pytorch-geometric.com/whl/torch-1.6.0/torch_sparse-latest%2Bcu102-cp38-cp38-linux_x86_64.whl
wget https://pytorch-geometric.com/whl/torch-1.6.0/torch_cluster-latest%2Bcu102-cp38-cp38-linux_x86_64.whl
wget https://pytorch-geometric.com/whl/torch-1.6.0/torch_scatter-latest%2Bcu102-cp38-cp38-linux_x86_64.whl
- 下载完成之后,在当前python 环境中
pip install *.whl 即可安装对应的whl 离线安装包
- 最后安装 torch-geometric
pip install torch-geometric
如果实在装不上 ,可以使用阿里云的
pip install torch-geometric==1.7.2 -i https://mirrors.aliyun.com/pypi/simple/
参考:https://zhuanlan.zhihu.com/p/138826540