No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.x‘
·
1、具体问题:执行编译相关命令行时,出现No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-11.x'
2、解决方案:
这是因为没有下载与cuda版本适配的torch导致的。(cuda版本不对,或者没有下载cuda版的torch)
3、具体解决方法:
1)nvcc -V 查看cuda版本
2)下载等于cuda版本的torch
pip install torch==<版本号>+cu<CUDA版本> torchvision==<版本号>+cu<CUDA版本> -f https://download.pytorch.org/whl/cu<CUDA版本>/torch_stable.html
例如
调整cuda为11.3后下载
pip install torch==1.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。
更多推荐


所有评论(0)