Linux下安装 torch_geometirc

最新推荐文章于 2024-08-08 19:56:43 发布

%3F%3F%B2

最新推荐文章于 2024-08-08 19:56:43 发布

阅读量527

点赞数

文章标签： pytorch cuda

本文链接：https://blog.csdn.net/qq_38809082/article/details/109249361

版权

本文指导如何在Linux操作系统中正确安装torch_geometric库，强调了pytorch、cuda版本匹配的重要性，并提供了一步步的安装步骤，包括检查现有pytorch和cuda版本，以及根据版本信息输入相应的安装命令。

摘要由CSDN通过智能技术生成

1、查看已安装的pytorch版本

import torch
print(torch.__version__)

2、查看安装的torch对应的cuda版本

>>> print(torch.version.cuda)

3、查看已安装的cuda版本

$nvcc --version

4、23步的版本要一致（重要）
5、依次输入以下命令

$ pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-geometric

（这里注意修改为自己对应的cuda版本和torch版本，如cuda10.1,torch1.5.0要改为
pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.5.0.html）
6、最后测试