我在执行python setup.py build develop的时候,编译出错,显示RuntimeError: Error compiling objects for extension
我在网上找了一些帖子,有让换pytorch版本的,也有其他方法的。我都试过,换pytorch版本确实能起效果,但是这个方法太笨了,我还是推荐使用另外一种方法。步骤如下:
1、进入cuda的安装目录
cd /usr/local
执行ls
,显示:
2、找到对应自己pytorch版本的cuda,比如我的有cuda、cuda-11.3、cuda-11.6,但是我的pytorch对应的是cuda(我把cuda11.1命名为cuda了)。
3、执行
vim ~/.bashrc
在最后面加上如下内容:
export CUDA_HOME=/usr/local/cuda
如果你的cuda版本为cuda-11.3的话,那应该加上如下内容:
export CUDA_HOME=/usr/local/cuda-11.3
然后退出vim,如果不知道怎么退出的话,可以按我说的操作,首先按"esc",然后输入":wq",最后按“enter(回车)”即可退出vim了。
4、执行如下代码,使得刚刚配置的source生效。文章来源:https://www.toymoban.com/news/detail-607274.html
source ~/.bashrc
5、此时执行python setup.py build develop,就可以成功编译了。文章来源地址https://www.toymoban.com/news/detail-607274.html
到了这里,关于RuntimeError: Error compiling objects for extension手把手带你解决(超详细)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!