torch-geometric包安装(适用于无法直接conda或pip安装)

1.确定你的torch版本cuda版本和python版本

python -V #terminal查看python版本
import torch

torch.__version__ #查看torch和cuda版本

2.去官网whl文件将包下载到本地

https://data.pyg.org/whl/torch-1.10.0+cu113.html

其中如果我的torch版本是1.10.0cuda版本是11.3。

如果你的torch版本是1.8.0cuda版本是11.1,只需要将上链接改为:

https://data.pyg.org/whl/torch-1.8.0+cu111.html

cpu的话,链接“+”号后面直接写cpu。

然后进去分别下载对应你版本的whl文件:

1.torch-scatter  2.torch-sparse  3.torch-cluster  4.torch-spline-conv  5.torch-geometric

3.pip本地安装

torch-geometric包安装(适用于无法直接conda或pip安装)_第1张图片

这里是我下载到本地的文件。

进入下载的目录,然后使用以下命令进行安装。

#注意以下安装顺序
pip install torch_scatter-2.0.4-cp37-cp37m-linux_x86_64.whl
pip install torch_sparse-0.6.1-cp37-cp37m-linux_x86_64.whl
pip install torch_cluster-1.5.4-cp37-cp37m-linux_x86_64.whl
pip install torch_spline_conv-1.2.0-cp37-cp37m-linux_x86_64.whl
pip install torch_geometric

注意,文件名的后缀需要修改为你本地的文件名!安装顺序要按照这个来!

你可能感兴趣的:(软件,深度学习,conda,pip,python)