编译的时候报错:no cuda runtime is found
首先确认torch版本是没问题的,
试了export FORCE_CUDA=“1” 并没啥用。
后来把换的conda源 还原回去,
在~/.bashrc中加入 export CUDA_HOME=/usr/local/cuda-11.X
souce ~/.bashrc
就解决了。
本文讲述了如何解决在编译过程中遇到的nocudaruntimeisfound错误,通过确认torch版本、环境变量设置和更改conda源,最终通过设置CUDA_HOME并source ~/.bashrc来修复问题。
编译的时候报错:no cuda runtime is found
首先确认torch版本是没问题的,
试了export FORCE_CUDA=“1” 并没啥用。
后来把换的conda源 还原回去,
在~/.bashrc中加入 export CUDA_HOME=/usr/local/cuda-11.X
souce ~/.bashrc
就解决了。
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1万+
2万+

被折叠的 条评论
为什么被折叠?