大连人工智能计算平台——华为昇腾AI平台——高性能计算HPC的pytorch源码编译报错——USE_CUDA=OFF——编译好的pytorch不支持CUDA的问题解决

发布时间 2023-08-09 19:37:46作者: Death_Knight

 

如题:

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

 

 

==========================================