大连人工智能计算平台——华为昇腾AI平台——高性能计算HPC的pytorch源码编译报错——USE_CUDA=OFF——编译好的pytorch不支持CUDA的问题解决
如题:
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
==========================================