当前位置:   article > 正文

【踩坑】一条指令解决torch_scatter等安装报错安装不上问题_安装torch-scatter

安装torch-scatter

转载请注明出处:小锋学长生活大爆炸[xfxuezhang.cn]

目录

背景说明

(推荐方法)解决方法一:使用conda安装。

解决方法二:指定pip的网站。

解决方法三:直接去下载whl文件。

(终极方法)解决方法四:配置MSVC

特殊情况说明


背景说明

        非常离谱,环境都是对的,但就是pip安装不上,像这样:


前提说明

建议先用 pip安装试试

pip install torch_scatter

如果是这个报错,就重装cuda对应的pytorch,不用往下看。

(推荐方法)解决方法一:使用conda安装

注意:此方法可能会有问题,注意备份环境后再操作~~~

           如果要稳妥的方法,可以直接看“解决方法四”!

  1. conda update -n base -c defaults conda
  2. conda install pytorch-scatter -c pyg

直接修好了!

解决方法二:指定pip的网站。

1、在pip安装时候,会显示你要装的版本是什么,比如:

2、使用pip安装时候指定网站

pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.2+${CUDA}.html

具体版本怎么填,可以看这个网站:torch-scatter · PyPI

解决方法三:直接去下载whl文件。

1、也是先知道要安装的torch_scatter的版本。

2、进这个网站:

https://data.pyg.org/whl/

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

3、选择你的pytorch版本,比如我的是pytorch2.1.2,并且我要装GPU版本,而且我的cuda是11.8版本的:

4、进去后,直接搜你想要的包,然后按照你的python版本操作系统选择,比如:

5、点击下载完成后,直接安装:

pip install .\torch_scatter-2.1.2+pt21cu118-cp39-cp39-win_amd64.whl

6、然后就可以用了

(终极方法)解决方法四:配置MSVC

如果以上方法都不行,那么只能按照提示来修复MSVC了。

1、根据提示安装这个

2、安装后打开软件

3、勾选MSVC的生成工具

或者直接勾选这个

4、然后pip重新安装即可

pip install torch-scatter

(可能要等很久...不过久说明正在编译安装,如果很快就出结果反而可能要报错)


特殊情况说明

        可能用conda安装完会报错,猜测大概率这里的dgl或pytorch不使用conda装的,出现了不兼容:

        可以考虑换解决方法二,或者卸载了库再用conda重装

  1. conda uninstall dgl pytorch torchvision torchaudio pytorch-scatter -y
  2. conda install dgl -c dglteam/label/cu118 -y
  3. conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia -y
  4. conda install pytorch-scatter -c pyg -y

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

闽ICP备14008679号