安装库报错:No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.3‘
短信预约 -IT技能 免费直播动态提醒
文章目录
1、报错内容
安装库时报错:
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-11.3'
2、检查
查看cuda版本和pytorch版本
python 进入python环境import torch torch.__version__torch.cuda.is_available()
nvidia-sminvcc -V
因此发现是由于该虚拟环境中CUDA与torch的版本不对应,发现在安装虚拟环境的environment.yml文件中只有指定cuda版本,而没有自动安装正确的torch版本:
channels: - pytorch - conda-forgedependencies: - python - cudatoolkit=11.3
3、解决
重新安装torch,在 torch 官网中找到适配的版本,并安装
pip install torch1.12.1+cu113 torchvision0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
注意:尽量使用pip安装,因为conda安装后有时还是不会安装cuda对应版本
来源地址:https://blog.csdn.net/weixin_41169280/article/details/129392977
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341