服务器安装pytorch_geometric torch_scatter踩坑记录

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
相关推荐
Empty_77714 小时前
Linux防火墙-Iptables
linux·运维·服务器
鸡哥爱技术14 小时前
Django入门笔记
笔记·python·django
saynaihe15 小时前
关于Ubuntu的 update造成的内核升级
linux·运维·服务器·ubuntu·devops
猫耳君15 小时前
汽车网络安全 CyberSecurity ISO/SAE 21434 测试之一
python·安全·网络安全·汽车·iso/sae 21434·cybersecurity
勘察加熊人15 小时前
python将pdf转txt,并切割ai
数据库·python·pdf
IMER SIMPLE15 小时前
人工智能-python-深度学习-神经网络VGG(详解)
人工智能·python·深度学习
Dersun16 小时前
python学习进阶之异常和文件操作(三)
开发语言·python·学习·json
Juchecar16 小时前
通过“单词补全”演示 Transformer 原理(Python代码可运行)
人工智能·python
c8i16 小时前
关于python中的钩子方法和内置函数的举例
python
apolloyhl16 小时前
深入理解 Linux 内核进程管理
linux·运维·服务器·操作系统