如题:
pytorch源码编译报错——USE_CUDA=OFF
在编译pytorch源码的时候发现错误,虽然编译环境中已经安装好CUDA和cudnn,环境变量也都设置好,但是编译好的pytorch包wheel总是在运行torch.cuda.is_available() 显示false,于是从编译源码的过程中进行重新检查,发现在编译的过程中提示:
USE_CUDA=OFF
-----------------------------------------------------------------
解决方法:
原先的CUDA路径为:
/usr/local/cuda-11.4
然后有一个软连接:
/usr/local/cuda =====>>>>>> /usr/local/cuda-11.4
我们将这个软链接删除掉,sudo rm /usr/local/cuda
然后将cuda11.4文件夹改名为:
sudo mv /usr/local/cuda-11.4 /usr/local/cuda
重新编译,显示成功配置,即:
USE_CUDA=ON
==========================================
标签:编译,pytorch,cuda,usr,local,CUDA From: https://www.cnblogs.com/devilmaycry812839668/p/17617839.html