一条命令解决安装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 的版本
相关推荐
别动我齐刘海2 分钟前
ROS2 Jazzy + C++ 实战路线——进阶学习3
c++·人工智能·vscode·python·算法·机器学习·机器人
Logintern095 分钟前
“以一致的顺序获取锁“是预防死锁最核心、最有效的手段
python·postgresql·锁机制
打工仔折腾 AI27 分钟前
Prometheus 告警推钉钉:从单群 Webhook 到跨网络 Alertmanager 实战
人工智能·后端·python·性能优化
Ivanqhz32 分钟前
Ping-Pong 双缓冲
开发语言·人工智能·python·深度学习·mlir
三十岁老牛再出发32 分钟前
9月18日总结
python·深度学习·机器学习
就叫你天选之人啦44 分钟前
安装torch+vllm+flash_attn的prompt
人工智能·pytorch·python
wuyk5551 小时前
21.计数排序:不用比较的“空间换时间”排序算法
python·算法·排序算法
智码看视界1 小时前
第4篇:循环神经网络及其变体 LSTM/GRU 实战
python·自然语言处理·gru·lstm·循环神经网络·情感分析·梯度消失
Web3&Basketball1 小时前
Agent外传审计实战:3类失准事故拦截脚本
python·大模型·agent·性能调优·推理优化
何以解忧,唯有..1 小时前
Milvus 向量数据库的 DDL、DML、DQL 操作详解
python