一条命令解决安装torch_scatter torch_sparse torch_cluster torch_spline_torch-cluster pip命令安装命令

作者：菜鸟追梦旅行 | 2024-04-20 18:40:39

踩

torch-cluster pip命令安装命令

在相应环境下输入

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
1

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

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

import torch
print(torch.__version__)  # 查看 torch 的版本
print(torch.version.cuda)  # 查看 CUDA 的版本
1
2
3

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】