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-smi
nvcc -V
因此发现是由于该虚拟环境中CUDA与torch的版本不对应,发现在安装虚拟环境的environment.yml文件中只有指定cuda版本,而没有自动安装正确的torch版本:
channels:
- pytorch
- conda-forge
dependencies:
- 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文章来源:https://www.toymoban.com/news/detail-519209.html
注意:尽量使用pip安装,因为conda安装后有时还是不会安装cuda对应版本文章来源地址https://www.toymoban.com/news/detail-519209.html
到了这里,关于安装库报错:No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.3‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!