一条命令解决安装torch_scatter torch_sparse torch_cluster torch_spline

在相应环境下输入

powershell 复制代码
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+cu118.html

该命令安装了pyg_lib包以及一些与PyTorch相关的包(torch_scatter、torch_sparse、torch_cluster、torch_spline_conv)从特定的包索引URL。指定的PyTorch版本是2.0.0,使用的CUDA版本是11.8。

如果你的版本不是我这样子的,可以输入以下指令查看自己 torch 和 cuda 的版本,然后修改上面那条命令后面的版本数字即可

python 复制代码
import torch
print(torch.__version__)  # 查看 torch 的版本
print(torch.version.cuda)  # 查看 CUDA 的版本
相关推荐
曲径海盗34 分钟前
【Python 学习第 38 天】for 循环
python·学习笔记·编程入门
武天旭34 分钟前
AI安全--AI大模型的后门攻击检测与防御
人工智能·深度学习·安全·机器学习
在世修行1 小时前
智学文枢:变式题生成引擎
python·ai·pyside·题目生成
On The Road :3 小时前
Java 多线程详解:从入门到线程安全
java·python·安全
岁月宁静3 小时前
如何设计支持 Skills 的 Agent?一套可扩展的四层架构
python·agent
岁月宁静3 小时前
CrewAI 快速上手:10 分钟搭建你的第一个 Agent 团队
python·agent
曲径海盗3 小时前
【Python 学习第 40 天】for 循环
python·学习笔记·编程入门
专注仿真4 小时前
知识问答大模型-rag算法详解
人工智能·python·控制算法
染指11104 小时前
158.Agent-LangGraph-子图(共享状态和不共享状态模式)
开发语言·人工智能·python·ui·agent·agents