当前位置: 首页 > news >正文

安装库报错:No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.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-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

http://www.lryc.cn/news/33568.html

相关文章:

  • CVTE前端面经(2023)
  • 基于EB工具的TC3xx_MCAL配置开发02_ICU模块配置
  • jmeter高阶系列--beanshell返回值中提取参数
  • 面向对象
  • mpi4py 运行过程中出现Read -1, expected xxx, errno = 1 解决方案
  • PMP考前冲刺3.07 | 2023新征程,一举拿证
  • 60条Python日常工作中的高频写法,收藏
  • (小甲鱼python)函数笔记合集七 函数(XI)总结 python函数的函数文档、类型注释、内省详解
  • Leetcode是什么
  • 2023-03-07 MySQL—基于规则优化-子查询优化
  • Rocketmq技术详解
  • TeeChart VCL/FMX v2023 crack
  • [Java·算法·困难]LeetCode32. 最长有效括号
  • pytorch如何搭建一个最简单的模型,
  • JS实现css的hover效果,兼容移动端
  • 企业微信的后台怎么进入和管理?
  • 【2223sW2】LOG2
  • buuctf-web-[SUCTF 2018]MultiSQL1
  • GitLab创建仓库分配权限
  • 代码随想录-51-110.平衡二叉树
  • 项目实战典型案例27——对生产环境以及生产数据的敬畏之心
  • 如何查找你的IP地址?通过IP地址能直接定位到你家!
  • Containers--array类
  • LinqConnect兼容性并支持Visual Studio 2022版本
  • 流量监管与整形
  • 详解init 容器
  • RequestResponseBodyMethodProcessor
  • 函数的极限
  • dnf命令使用
  • CLIP CLAP