编译的时候报错:no cuda runtime is found

首先确认torch版本是没问题的,
试了export FORCE_CUDA=“1” 并没啥用。

后来把换的conda源 还原回去,
在~/.bashrc中加入 export CUDA_HOME=/usr/local/cuda-11.X
souce ~/.bashrc
就解决了。

参考链接

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐