赞
踩
torch-geometric
需要提前安装:torch-scatter
、torch-sparse
、torch-cluster
、torch-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
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
注:
他这里提到的是通过这种下载方式
把${CUDA}
换成 (cpu
, cu102
, cu113
, cu115
) ,把 ${TORCH}
换成 (1.10.0
, 1.11.0
)。
其他的版本,比如我想要一个pytorch
是1.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.
注:
其实这就是上面提到的匹配(torch
和cuda
版本要对应,可以根据pytorch
官网进行查看,其实这是最开始安装pytorch
的时候应该考虑的问题,这个下面在讲一下吧 ),以及要符合现在torch-geometric
中下载库的要求(上面的下载方式当前只支持CUDA
: (cpu
, cu102
, cu113
, cu115
),TORCH
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。