python安装torch-geometric库
发表于:2023-02-20 |
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric
上面是别人提供的安装方法,这里${CUDA} 和${TORCH} 是你的torch和相应的CUDA版本,比如我的torch版本是1.8.0 CUDA版本是11.1,安装torch-scatter的指令就是

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html

但是按照上述方法操作会出现问题,大概意思是说你的gcc版本不对(千万不要被这个误导!!!)

网上提供的解决方案要么毫无作用,要么就是安装后import时又会报错,

虽然并未完全解决我的问题,但却为问题解决提供了一些思路(版本依赖问题)

其实可以先看一下安装指令 -f 后面的链接(我的torch及cuda版本对应的链接即https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html)里到底有什么,打开后发现是一些轮子资源

于是我发现这样一个问题,我安装的torch-scatter默认版本是2.0.9,如下图(因为我下载过了所以有cache),但在上面的链接中发现并没有对应的版本,于是我想可能是这里出了问题

将torch-scatter的版本改为上面的链接中包含的版本(2.0.6~2.0.8),这里我改成2.0.8,再次pip install,安装成功!(如下图)

按这个思路依次安装其他几个包,最后再安装torch-geometric,我执行的指令依次是:

pip install torch-scatter==2.0.8 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-sparse==0.6.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-geometric

均安装成功,最后试一下import,正常!

上一篇:
西安交通大学软件镜像站Pypi 镜像使用帮助
下一篇:
ImportError: libgthread-2.0.so.0: cannot open shared object file: No such file or directory