torch_scatter安装踩坑实录及解决方法(ubuntu/conda/python/torch/cuda/torch_cluster)

【简洁版解决方案直接跳到最后】

一。安装途径

一般在ubuntu中配置python环境,选择conda安装是不错的选择。但是使用anaconda网站上torch_scatter的安装指令conda install -c esri torch-scatter安装后可能会发现无法成功运行,而且安装过程中还会修改你其他库的版本,导致环境崩掉。这是因为当你没有指定安装库的版本时, conda会从当前安装渠道寻找最新版本的安装包,并且自动将你已经安装的其他环境更新到和它匹配。经过对使用torch_scatter的开源项目的查询我发现基本上在requirement这一栏都会建议使用pip来安装torch_scatter。在使用pip安装时,如果没有指定版本号,则 pip会根据你已有的环境,自动安装和你当前环境最匹配的版本。因此,我们建议使用pip安装torch相关库。使用anaconda时注意在anaconda中create虚拟环境后在虚拟环境内pip安装。

二。版本问题

有的开源项目会给出他们使用的torch_scatter的pip install链接,但是由于各种库的版本不断更新,安装后很可能出现和torch的cuda版本不匹配的问题。

如果你在本站搜索这个问题的解决方法,十有八九会告诉你,先查看自己的torch版本,然后去下载对应版本的torch_scatter的whl文件并安装。你可以试一试看看是否成功。如果发现不成功请往下看。

首先要明确的一点是,我们需要保持torch和torch_scatter版本的高度一致。经过第一部分的叙述我们可以知道,使用pip来安装是比较好的选择。

首先我们直接去https://download.pytorch.org/whl/torch_stable.html下载torch的whl文件并用pip install 安装。方法一:在网站找到你需要的whl,右键,复制链接,使用【wget 链接】进行下载。下载后使用【pip install 文件名】进行安装。方法二:复制链接后,直接【pip install 链接】进行安装。 这里建议使用方法一,并注意查看下载的whl文件是否是cuda版本!在网站页面寻找链接时,不要看左侧括号内的cuda版本,那个好多是错的,要看文件名中间是否有+cu10x的字样!例如,我的python版本是3.8,cuda版本10.2,那么我可以安装torch-1.9.0+cu102-cp38-cp38-linux_x86_64.whl。但是torch-1.7.0和1.8.0虽然网页上左侧写了cu100、cu101,下载下来后会发现文件名没有cuda版本,如果你是cu10.0或10.1,安装会报错。

【简洁版总结】安装torch及其相关库不成功时,建议全部使用pip安装,如果正在使用anaconda,则在conda虚拟环境内使用pip安装。

torch,torchvision,torchaudio此三样通过在http://download.pytorch.org/whl/torch_stable.html下载指定版本的wheels并pip安装(注意检查下载下来的文件名中是否有cuda版本号,没有则可能不是cuda版本)。

torch_cluster,torch_scatter,torch_sparse则在成功安装torch后,使用指令pip install –no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-1.x.0+cu10x.html(torch_cluster可直接修改为另外两样)进行安装即可。

如果本文章对你有用,或你有任何质疑,可以在评论区留言。

Original: https://blog.csdn.net/weixin_43980074/article/details/124137375
Author: hmsgsfz
Title: torch_scatter安装踩坑实录及解决方法(ubuntu/conda/python/torch/cuda/torch_cluster)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/770438/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球