我在执行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生效。

source ~/.bashrc

5、此时执行python setup.py build develop,就可以成功编译了。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐