第一步:在Pytorch中下载需要的版本
链接:https://pytorch.org/get-started/previous-versions/
e.g. torch == 1.7.1
1 2 3 4 5 6 7 8 9 10 11 12 13 14
| pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
|
第二步:根据自己下载的Torch、cuda、python版本,下载对应的torch-geometric, torch-sparse, torch-scatter, torch-cluster, torch-spline-conv。
链接:https://data.pyg.org/whl/
e.g.
1 2 3 4 5 6 7
|
pip install torch_cluster-1.5.8-cp37-cp37m-linux_x86_64.whl pip install torch_scatter-2.0.5-cp37-cp37m-linux_x86_64.whl pip install torch_sparse-0.6.8-cp37-cp37m-linux_x86_64.whl pip install torch_spline_conv-1.2.0-cp37-cp37m-linux_x86_64.whl pip install torch_geometric==1.7.0
|
完成!