服务器安装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
相关推荐
Liang_GaRy3 分钟前
心路历程-三个了解敲开linux的大门
linux·运维·服务器
wyiyiyi7 小时前
【Web后端】Django、flask及其场景——以构建系统原型为例
前端·数据库·后端·python·django·flask
mit6.8247 小时前
[1Prompt1Story] 滑动窗口机制 | 图像生成管线 | VAE变分自编码器 | UNet去噪神经网络
人工智能·python
没有bug.的程序员7 小时前
JVM 总览与运行原理:深入Java虚拟机的核心引擎
java·jvm·python·虚拟机
玩转以太网7 小时前
基于W55MH32Q-EVB 实现 HTTP 服务器配置 OLED 滚动显示信息
服务器·网络协议·http
甄超锋8 小时前
Java ArrayList的介绍及用法
java·windows·spring boot·python·spring·spring cloud·tomcat
小晶晶京京8 小时前
day34-LNMP详解
linux·运维·服务器
画个太阳作晴天8 小时前
A12预装app
linux·服务器·前端
AntBlack8 小时前
不当韭菜V1.1 :增强能力 ,辅助构建自己的交易规则
后端·python·pyqt
碎像10 小时前
Linux上配置环境变量
linux·运维·服务器