IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> pytorch无法调用gpu的一些解决方法 -> 正文阅读

[人工智能]pytorch无法调用gpu的一些解决方法

出现错误:

1.UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version

2.导入torch后torch.cuda.is_available()为false

出现原因:

最近服务器重启,重启之后驱动版本竟然降了(玄!!!)。降了之后发现pytorch又调用不了gpu了,但是这次又不能升级nvidia驱动版本(唉就不多说了,大家如果不明白也可以看看我上一篇文章,也是关于如何解决这个问题的)。

目前的环境

驱动版本跟真实安装的版本统一为10.1,按理来说驱动版本大于等于真实版本并不会出问题,确实不会是这样,出现问题的原因是cudatoolkit版本的版本装错了,安装的cudatoolkit版本必须小于驱动版本。如果未解决你的疑惑,可以参考下面的文章。显卡驱动版本、cudatoolkit版本、cudann版本、tensorflow-gpu版本之间的对应关系_Y.G.bh的博客-CSDN博客_cuda和cudatoolkit版本

进入pytorch官网PyTorch

?我目前是驱动版本为10.1,我于是将这个设为cudatoolkit=10.1(在我上一篇文章2021(接近2022)linux(ubuntu16.04)安装cuda跑深度学习步骤以及踩坑过程(ELF标头无效、导入torch后torch.cuda.is_available()为False等等)_Dcityy的博客-CSDN博客有提到),但因为我之前的驱动版本是11.4所以不会造成影响,现在版本降低就出现了问题。所以正确的做法是将cudatoolkit设置为小于10.1的。但这里又需要注意一个版本对应问题。

版本对应

根据这篇文章显卡驱动版本/Cuda版本/Tensorflow/Pytorch版本的对应关系_#欧吼的博客-CSDN博客_tensorflow对应显卡版本

?我们看到10.1对应的版本有很多,接下来我们进入pytroch官网查看历史版本Previous PyTorch Versions | PyTorch

?进入后可以在左上角看到有vx.x.x

?在这里我根据版本对应的关系挑选了v1.5.1,并且cudatoolkit=9.2(小于10.1)的版本

?

安装

conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch

命令之后就解决了。

?

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2022-04-18 17:43:23  更:2022-04-18 17:46:15 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 10:47:20-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码