torch_geometric可以命令模式安装方法
注意:每个python虚拟环境都是可以安装此环境独自需要的cuda版本等。
一、首先在python程序运行的过程中出现了“RuntimeError: Not compiled with CUDA support”错误。
二、然后经过博主不断尝试网上各种解决方案都没有解决,然后发觉可能是之前安装的torch_scatter文件出现了问题(之前安装参考文章和方式如下,也就是直接下载torch_scatter对应版本的whl文件然后安装)。
pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
参考安装文章:点击打开《基于Pytorch中安装torch_geometric简单详细完整版》文章
三、然后博主决定重新更换torch_scatter版本,但是出现“OSError: /usr/local/cuda-10.2/lib64/libcudart.so.10.1: version `libcudart.so.10.1’ not found (required by /home/xieru21/anaconda3/envs/hiv/lib/python3.7/site-packages/torch_scatter/_scatter_cuda.so)”错误,说明更换torch_scatter版本是有问题的。然后博主又重新和之前安装方式一样重新用whl文件安装torch_scatter,但是还是出现“RuntimeError: Not compiled with CUDA support”错误。
四、然后本人决定更换安装方式,采用命令形式不用whl文件安装,注意:博主之前在此python虚拟环境安装的就是cuda10.2和pytorch1.10,所以下面的命令中采用这个torch-1.10.0+cu102,否则需要更改,链接前面的内容可以不更改只需要更改“torch-1.10.0+cu102”此部分,怎么更改除参考自己虚拟环境的cuda和pytorch版本,还需要点击此链接核对一下对应的cuda和pytorch版本。比如:你的虚拟环境pytorch版本是1.11.0,cuda版本是11.3那么命令中的链接https://data.pyg.org/whl/torch-1.10.0+cu102.html改成https://data.pyg.org/whl/torch-1.11.0+cu113.html,其他照旧。安装完成之后发现torch_scatter版本还是之前安装的版本,说明之前安装的方式还是有些bug存在。
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
y
五(可跳)、鉴于之前的安装方式有bug存在,那么基于Pytorch中安装torch_geometric可以换成新的命令模式安装,这个安装方式是和此篇《基于Pytorch中安装torch_geometric简单详细完整版》文章对比。博主举例的依旧是cuda版本是10.2,然后pytorch版本是1.10。
- 创建一个叫做test的python=3.7的虚拟环境
conda create -n test python=3.7
y
- 激活test虚拟环境
conda activate test
- 安装cuda10.2(注意:如果conda版本等级太低是无法用conda命令安装cuda高级版本的,比如cuda11.3.1,所以需要先升级conda版本,命令是:
conda update -n base -c defaults conda
)
conda install cudatoolkit==10.2.89
y
- 下载torch1.10版本,注意:Pytorch各版本下载网页:点击打开链接
pip install torch-1.10.0+cu102-cp37-cp37m-linux_x86_64.whl
- 下载配置torch-geometric,注意:torch_sparse的安装还是要指定其安装版本,否则会报错安装失败,但是torch-geometric可以不指定安装版本,但是根据需求可以指定。
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
pip install torch_sparse==0.6.13 -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
pip install torch_cluster -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
pip install torch_spline_conv -f https://data.pyg.org/whl/torch-1.10.0+cu102.html
pip install torch-geometric==2.0.1
- 测试torch_geometric安装是否成功,复制下面命令运行没有报错就说明安装成功。
python
import torch_geometric