当前位置:   article > 正文

torch-geometric安装教程及常见源下载报错问题_torch-geometric=1.5.0就是安不好

torch-geometric=1.5.0就是安不好

安装 torch-geometric

一、介绍

需要提前安装:torch-scattertorch-sparsetorch-clustertorch-spline-conv
安装的时候会遇到很多的问题,版本号以及安装包源的问题

安装框架模型,大约装了几个之后,每次给的教程都是会报错,就像是中了魔咒一样。
其实最开始就感觉,安装东西还是要根据官方网站来解决这些问题。
所以我们就参考当时的官方网站的安装教程。

官方安装教程

二、教程

我们在官方教程里面主要是看Installation via Pip Wheels部分,使用Anaconda进行安装虚拟环境

Install the relevant packages部分
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric
  • 1
  • 2
  • 3

where ${CUDA} and ${TORCH} should be replaced by the specific CUDA version (cpu, cu102, cu113, cu115) and PyTorch version (1.10.0, 1.11.0), respectively. For example, for PyTorch 1.11.* and CUDA 11.3, type:

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+cu113.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.11.0+cu113.html
pip install torch-geometric
  • 1
  • 2
  • 3

注:

  1. 他这里提到的是通过这种下载方式${CUDA} 换成 (cpu, cu102, cu113, cu115) ,把 ${TORCH}换成 (1.10.0, 1.11.0)。

    其他的版本,比如我想要一个pytorch1.6.0的,那么就不符合人家的条件了,就不能这样下载了,需要看一下面啦!

**重点**

Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1 and PyTorch 1.9.0 (following the same procedure). For older versions, you need to explicitly specify the latest supported version number in order to prevent a manual installation from source. You can look up the latest supported version number.

注:

  1. 其实这就是上面提到的匹配(torchcuda版本要对应,可以根据pytorch官网进行查看,其实这是最开始安装pytorch的时候应该考虑的问题,这个下面在讲一下吧 ),以及要符合现在torch-geometric中下载库的要求(上面的下载方式当前只支持CUDA: (cpu, cu102, cu113, cu115),TORCH

    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/623529
推荐阅读
相关标签