一条命令解决安装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 的版本
相关推荐
只会安静敲代码的 小周5 小时前
AI应用开发
python
txg6665 小时前
Agent 攻击 Agent:自动检测 LLM Agent 中的污点式漏洞
人工智能·深度学习·安全·网络安全
FriendshipT5 小时前
Ultralytics:解读C3Ghost模块
人工智能·pytorch·python·深度学习·目标检测
Black蜡笔小新5 小时前
企业AI算力工作站DLTM深度学习推理工作站AI视觉检测助力医疗影像分析
人工智能·深度学习·视觉检测
Java面试题总结5 小时前
Vue3流式调用大模型接口完整实践
后端·python
怪兽学LLM5 小时前
AI Agent 记忆系统设计:长短期记忆如何实现?什么时候存?什么时候查?
人工智能·python
DFT计算杂谈5 小时前
DeepSeek 集群服务器无root本地部署指南
数据库·人工智能·python·opencv·算法
卷无止境5 小时前
Python 跨平台 GUI 解决方案全景报告
前端·python
benchmark_cc15 小时前
如何用 Python + QuantDash 快速构建高胜率“配对交易(Pairs Trading)”策略?
开发语言·人工智能·python·pandas·量化交易·quantdash
Python+9916 小时前
Java 枚举类(Enum)详解:从基础到高级应用
java·开发语言·python