torch-geometric手动安装

最新推荐文章于 2024-08-21 19:53:25 发布

布布可爱熊

最新推荐文章于 2024-08-21 19:53:25 发布

阅读量373

点赞数

分类专栏：工具安装问题文章标签：深度学习 pytorch 计算机视觉

本文链接：https://blog.csdn.net/qq_41520765/article/details/131811928

版权

本文介绍了如何手动安装torch-geometric库，包括检查torch和cuda版本，寻找匹配的安装包，通过conda逐个安装依赖，以及不下载直接从url安装的方法。

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

1. 查看torch版本和cuda版本，具体代码如下：

# torch版本
conda list中查看
# cuda版本
import torch
print(torch.version.cuda)

2. 在下面的网站查找有没有与上面两个版本相对应的安装包

https://pytorch-geometric.com/whl/

3. 如果有的话，有两种方法可以安装

3.1 手动下载依赖，逐个安装，推荐的几种可行的配置

python=3.8
           torch==1.8.0+cu111
           torch-scatter==2.0.6
           torch-cluster==1.5.9
           torch-sparse==0.6.9
           torch-geometric==1.6.3

python=3.7

           torch==1.1.0
           torch-scatter==1.3.2
           torch-cluster==1.4.5
           torch-sparse==0.4.3
           torch-spline-conv==1.1.1
           torch-geometric==1.3.2
           torchvision==0.3.0

python==3.6

           torch==1.3.0
           torch-scatter&