在执行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了