在keras中使用gpu加速训练模型,如何安装cuda,cudnn,解决cudnn_cnn_infer64_8.dll 不在path中,解决device_lib.list_local_devices()中无gpu,解决jupyter使用gpu训练总是挂掉的等问题

全网最全!!!亲自踩坑

首先打开任务管理器,查看自己电脑是否存在gpu

1、安装tensorflow-gpu

Pip install tensorflow-gpu==2.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --user

注意,以前安装的tensorflow默认是cpu版的,所以我们下载这个是gpu版的,注意版本得和tensorflow一致(网上说要在tensorflow安装之后安装tensorflow-gpu)

2、安装keras (先安装keras的话,安装tensorflow会把karas安装到一致的版本)

3、安装CUDA

首先在命令行输入 NVIDIA-smi,查看自己的gpu,驱动版本,CUDA版本

打开网址 https://developer.nvidia.com/cuda-toolkit-archive
按照以下表格安装对应的CUDA版本(我的11.5是已经下载好的),下载exe

下载这个东西,然后双击下载好的东西
注意:在装CUDA时选择自定义安装,取消“Visual Studio”这一项就可以了(大概在第一个下面)

4、安装cudnn

在这个网址,下载对应的cudnn:[https://developer.nvidia.com/rdp/cudnn-archive](https://developer.nvidia.com/rdp/cudnn-archive)

比如我下载的就是这个

将下载好的压缩包解压,然后只需要把下载后的压缩文件解压缩,分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5对应的include、lib、bin目录下即可。

5、添加环境变量

把安装的CUDA的bin目录和libnvvp加到path环境中

6、测试代码

首先用以下代码,查看自己的设备信息

from tensorflow.python.client import device_libdef get_available_gpus():local_device_protos = device_lib.list_local_devices()print(local_device_protos)return [x.name for x in local_device_protos if x.device_type == 'GPU']
get_available_gpus()

  • 如果没有gpu说明安装的cuda版本不对,找不到电脑的gpu,需要在控制面板里面把CUDA开头的之前安装的版本的文件都卸载,再把C:\Program
    Files\NVIDIA GPU Computing Toolkit全部删除,重复以上步骤安装对应的CUDA版本

  • 如果以上代码运行发现有gpu了,说明版本没错,现在可以测试一下自己的代码了,通过一下代码指定使用gpu进行训练,直接开跑即可

如果弄好之后运行jupyter文件,每次都挂掉了,打开jupyter的命令行窗口查看,是否存在
Please make sure cudnn_cnn_infer64_8.dll is in your library path!
这句话,如果存在,首先检查自己的环境变量中是否在path中添加了这两个,没有的话添加进去

如果添加了还是这种情况,那么就是电脑中没有zlibwapi.dll文件。
打开这个网址
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-zlib-windows

点击安装这个,解压下载的压缩包,打开dll_x64文件夹

复制这个dll文件,将它复制到C:\Windows\System32目录下,重新启动jupyter,发现已经可以运行了

注意:multi_gpu_model这个方法是多个gpu运行才需要指定的,单个gpu直接开跑即可

参考文章:
https://blog.csdn.net/fun_always/article/details/103357840
https://www.jianshu.com/p/9bdeb033e765
https://blog.csdn.net/weixin_44704985/article/details/109500188
https://blog.csdn.net/qq_46126258/article/details/112708781

在keras中使用gpu加速训练模型;安装cuda;cudnn;cudnn_cnn_infer64_8.dll 不在path中;device_lib.list_local_devices无gpu;挂掉相关推荐

  1. 安装cuda+cuDNN

    安装cuda+cuDNN 一.概述 二.安装GPU驱动的要点 三.成功安装的样例 样例1. Win10+CUDA-v10.1 安装环境描述 安装步骤 样例2. Ubuntu20.04+CUDA-v11 ...

  2. 从零点一开始机器学习之Win10 64位下安装Cuda+Cudnn

    系列文章目录 从零点一开始机器学习之晦涩难懂的各种概念 从零点一开始机器学习之Win10 64位下安装Cuda+Cudnn 从零点一开始机器学习之GPU运算性能和CPU性能对比(GPU算力表) 从零点 ...

  3. 安装Nvidia驱动并使用Anaconda搭建GPU环境(安装CUDA)+ PyTorch + TensorFlow + PyTorch Lightning

    前言: 准备在实验室的一个服务器上用GPU的,但是这个服务器好久没有人用了,cuda还是10.0版本,需要重装cuda,嫌麻烦,然后发现Nvidia官网居然有anaconda安装CUDA的教程,所以考 ...

  4. ubuntu16.04下安装GTX1080TI显卡驱动+安装CUDA+Cudnn+anaconda+GPU版tensorflow

    参考博客:https://blog.csdn.net/weixin_40294256/article/details/79157838 先安装好GTX1080Ti 显卡驱动--安装CUDA--安装Cu ...

  5. Win10搭建GPU版Darknet——VS2019+CUDA+CUDNN篇

    一.理论知识简介 为了更好的理解gpu必要性,参考了一位博主的介绍,并对主要内容进行总结如下(如有要详细了解原理的,文末放置原博主链接): 1.1 CPU与GPU的区别 CPU和GPU不同之处在于其设 ...

  6. cuda nvcc版本不一致_入坑第一步:Win10安装cuda+cuDNN+TensorFlow-GPU走过的那些路

    这两天安装tensorflow-gpu被折腾够呛,幸亏最后成功了,给想要安装的大神们看下我走过的坑,避免掉入. 如果是新手,需要安装下面几个软件: 第一步 安装anaconda 首先就是Python编 ...

  7. Ubuntu 20.04安装CUDA CUDNN 手把手带你撸

    新手先看这 之前一直使用CPU做训练,最近手上多了台单卡1080Ti显卡主机,于是开始研究GPU训练.用GPU训练一定会使用CUDA了,刚开始接触CUDA非常非常头痛,对小白很不友好,我几乎整了整整一 ...

  8. tensorflow GPU环境配置 Nvidia+cuda+cudnn

    显卡安装 去官网下载显卡驱动https://www.nvidia.cn/ 删除旧的nvidia驱动(刚装的机子不需要这一步) sudo apt-get purge nvidia* sudo chmod ...

  9. Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch

    步骤: 1.安装显卡驱动: 2.安装CUDA: 3.安装CUDNN: 4.安装Conda: 5.安装Pytorch. 一.系统和硬件信息 1.Ubuntu 20.04 2.显卡:1050Ti 二.安装 ...

最新文章

  1. oracle导出对象和数据,浅析Oracle对象和数据的导入导出
  2. 房价波动5%很正常 房地产市场绝不会崩盘
  3. php表单数据提交到本业,PHP_PHP+Mysql+jQuery实现发布微博程序 php篇,先还是要说明本例的业务流程 - phpStudy...
  4. 10上wsl位置迁移_wsl的安装/升级笔记
  5. SAP Fiori 应用 My Appointment - Belonging to me, Search by team, Search by group
  6. 串的定长存储表示【数据结构】
  7. esp8266原理图_ESP32/ESP8266使用MicroPython控制DHT11/DHT22
  8. 垃圾回收算法与实现系列-JVM无锁实现
  9. 计算机二级是报office还是vb,请问一下计算机二级中,高阶office好考,还是VB好考?...
  10. Got error ‘parentheses not balanced‘ from regexp
  11. 系统、驱动相关软件下载
  12. 酒店管理系统需求分析
  13. PowerDesigner修改设置表格背景色
  14. 什么是六类网线和超六类网线?
  15. android 监听锁屏 权限,Android中监听锁屏变化和防止锁屏
  16. xtrabackup全量+增量备份
  17. hdu 5535 Cake 构造+记忆化搜索
  18. 如何查询SCI和EI索引号
  19. java double 赋值语句_Java中float、double、long类型变量赋值添加f、d、L尾缀问题
  20. python打开.mat文件_python怎么打开mat文件?

热门文章

  1. VS code 左侧菜单栏不见了/只能打开一个编辑页面
  2. Discord/MidJourney注册遇到电话号码无效 invalid phone number
  3. RedHat RHEL7.2 系统安装详细步骤
  4. Armadillo | 复数小记
  5. 计算机大赛剪辑的视频,视频剪辑大赛
  6. python,java,go语言之间的区别!
  7. linux系统如何下游戏,海岛纪元干货 在Linux系统下如何畅玩游戏攻略
  8. 生活有时候还是需要点这个的
  9. 华为云云筑·开发者年度盛典精彩回顾
  10. 图片压缩 in Android