服务器安装pytorch_geometric torch_scatter踩坑记录

我的心永远是冰冰哒2024-09-24 15:14

bash 复制代码

conda create -n pyg python==3.8.12
pip install torch==1.13.0

安装的版本如下

bash 复制代码

pip install torch-scatter
pip install torch-sparse
pip install torch-cluster
pip install torch-spline-conv
pip install torch-geometric==2.2.0
pip install ipykernel
python -m ipykernel install --name pyg --user

此时会出现

在torch_scatter包中，出现

python scatter_max算子不适配报错，RuntimeError: Not compiled with CUDA support的错误

解决办法如下

torch scatter的版本如下

bash 复制代码

ERROR: Could not find a version that satisfies the requirement torch-scatter==2.0.1 (from versions: 0.3.0, 1.0.2, 1.0.3, 1.0.4, 1.1.0, 1.1.1, 1.1.2, 1.2.0, 1.3.0, 1.3.1, 1.3.2, 1.4.0, 2.0.2, 2.0.3, 2.0.4, 2.0.5, 2.0.6, 2.0.7, 2.0.8, 2.0.9, 2.1.0, 2.1.0+pt113cu117, 2.1.1, 2.1.1+pt113cu117, 2.1.2)

默认选择安装torch_scatter2.1.2

请使用下面的命令就可以卸载并重新安装torch_scatter包了，这样就可以愉快的跑代码了

bash 复制代码

pip uninstall torch_scatter
pip install torch-scatter==2.1.1 -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html --no-cache-dir

本文是转载文章，点击查看原文

如有侵权，请联系 xyy@jishuzhan.net 删除