一条命令解决安装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 的版本
相关推荐
Warson_L2 小时前
Python `Annotated` 与 LangGraph Reducer 学习笔记
python
韩师傅2 小时前
海天线算法的前世今生
python·计算机视觉
韩师傅2 小时前
当你的甲方设备过烂,要如何快速出效果?
python·计算机视觉
Warson_L2 小时前
LangGraph的MessageState and HumanMessage
python
韩师傅3 小时前
当你的甲方吐槽天空不够蓝,你应该如何应对
python·计算机视觉
Warson_L4 小时前
python的类&继承
python
Warson_L4 小时前
类型标注/type annotation
python
ThreeS6 小时前
手搓MiniVLA全实战教程-一步一步用pytorch解释原理与思路
人工智能·python
金銀銅鐵7 小时前
[Python] 模 n 乘法的逆元计算器
python·数学·游戏