tensorflow+Cuda+cuDNN环境配置,开启GPU训练记录

本文记录了一位开发者配置Tensorflow-GPU环境的过程,包括选择合适的tensorflow、CUDA和cuDNN版本,以及解决安装过程中遇到的问题。重点强调了版本匹配的重要性,并提供了详细的操作步骤,如安装CUDA和cuDNN,配置环境变量,以及测试GPU是否成功启用。此外,还分享了安装CUDA时遇到的自动重启问题及其解决方案。
摘要由CSDN通过智能技术生成

记录一下自己配置过程。
我参考了这篇文章任意门,同时也有一些自己的经验教训总结如下:
1、首先要弄清楚自己电脑环境以及当前训练是否开启了GPU。
我用的是jupyternotebook + tensorflow2.5,cuda是11点几,我的gpu是1080,后来我把tensorflow和cuda卸载了,选择配置tensorflow-gpu 2.2 +cuda10.1 +cudnn7.65
后来自己配了一个电脑显卡是gtx-2060,cuda选择11.7,cudnn选择8.4.0 tensorflow无需特别安装gpu版,官网做了统一。

查看电脑的显卡
查看能否开启GPU训练

import tensorflow as tf
tf.config.experimental.list_physical_devices('GPU')

输出为[ ] 就说明没有开启GPU。
配置过程必须严格保证tensorflow版本,CUDA版本,cuDNN版本保持匹配!!!
这是tensorflow官网,有版本对应表格。传送门
tensoflow-gpu 2.2 CUDA 10.1 cuDNN7.5

2、安装tensorflow gpu版

pip install tensorflow-gpu==2.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

这里使用了清华园镜像。
3、安装CUDA
我下的是CUDA10.1
https://developer.nvidia.com/cuda-10.1-download-archive-base
当我安装11.7遇到的问题是安装到一半自动重启,重启后找不到nvcc.exe,我做了两件事,一个是安装的模式由精简换成了自定义,然后把组件全安装了,一个是他提示我没装visual studio,我给补上了,之后就正常了,不知道哪个步骤起的作用。
(注意版本!注意版本!注意版本!)
安装完命令行输入”nvcc -V”可以查看
4、安装cuDNN
传送门
这个网站访问非常慢,而且按照正常步骤下载需要账号密码,非常麻烦。按照网上的方法可以绕过登录,右键点击要下载的版本,复制链接,然后用迅雷等下载工具直接下载。
在这里插入图片描述
5、配置
在配置之前我还刻意测试了一下,还是无法启动GPU,说明仅仅安装好还是不行!
把cudnn里面的文件全部复制到这CUDA里。如图所示把上面图的内容全部复制到下面图的路径。
在这里插入图片描述
在这里插入图片描述
设置路径
在系统环境变量的Path里添加如下路径(根据自己实际路径调整):
在这里插入图片描述

6、测试
在jupyter notebook里:
在这里插入图片描述

  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值