解决RuntimeError: Error compiling objects for extension

发布时间:2024年01月23日

在执行python setup.py install的时候,编译出错👇

除了更换torch版本外,下面这个方法简便一些:

step1:

进入cuda安装目录

cd /usr/local

执行 ls,显示大概以下内容:

?找到自己的cuda版本,我的是cuda-11.3

step2:

编译环境:(nano也可以换成vim)

nano ~/.bashrc

打开后进入最后一行,添加下面的内容

export CUDA_HOME=/usr/local/cuda-11.3

这个cuda-11.3换成你们自己的

step3:

重启环境:

source ~/.bashrc

然后我再执行python setup.py install就OK了

文章来源:https://blog.csdn.net/2301_77554343/article/details/135771867
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。