安装库报错:No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.3‘
创始人
2024-05-30 11:41:51

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

因此发现是由于该虚拟环境中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

相关内容

热门资讯

苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
世界上最漂亮的人 世界上最漂亮... 此前在某网上,选出了全球265万颜值姣好的女性。从这些数量庞大的女性群体中,人们投票选出了心目中最美...
长白山自助游攻略 吉林长白山游... 昨天介绍了西坡的景点详细请看链接:一个人的旅行,据说能看到长白山天池全凭运气,您的运气如何?今日介绍...
猫咪吃了塑料袋怎么办 猫咪误食... 你知道吗?塑料袋放久了会长猫哦!要说猫咪对塑料袋的喜爱程度完完全全可以媲美纸箱家里只要一有塑料袋的响...