安装库报错：No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.3‘

文章目录

• • 1、报错内容
  • 2、检查
  • 3、解决

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