前言

安装Tensorflow-gpu 与 keras的时候,一定先要注意版本的对应,不然很容易出错,在看的时候,建议先看完整篇文章再上手。

一、环境+配置

本机环境

显卡:RTX3050Ti(notebook)

Windows10专业版

NVIDIA 511.65

网上查到的可行版本

(跟本人所使用的有所偏差)

python3.7.0+CUDA11.6.0+cuDNN8.3.2+tensorflow2.7.0+Keras2.7.0
python3.7.0+CUDA11.3.1+cuDNN8.2.1+tensorflow2.7.0+Keras2.7.0
python3.7.0+CUDA10.1.2+cuDNN7.6.5+tensorflow2.2.0+keras 2.3.1

本人所安装CUDA、CUDNN、tensorflow-gpu、keras 版本

cuda_11.6.1_511.6

cudnn_8.3.2.44

tensorflow-gpu 2.7.0

keras 2.7.0

即最终本人所用环境为

python3.8.12 + CUDA 11.6.1 + cuDNN 8.3.2 + tensorflow 2.7.0 + Keras 2.7.0

tensorflow与CUDA、cuDNN关系查询

https://tensorflow.google.cn/install/source_windows?hl=en#gpu

CUDA版本选择

NVIDIA 与CUDA的版本对应关系

参照表格选择

首先,在桌面【右键】-打开【NVIDIA 控制面板】-选择【帮助】-【系统信息】,再打开的系统信息中选择【组件】,即可看到本机GPU对应支持的CUDA版本:

组件3D设置重点NVCUDA64.DDL 后面的产品名称

或者可以在终端输入命令nvidia-smi查看GPU驱动版本

CUDNN版本选择

CUDA与CUDNN的版本对应关系

根据CUDA的版本进行安装,下载压缩包版本

二、安装教程

2.1 CUDA安装

双击cuda_11.6.1_511.65_windows.exe,进入安装程序:

这里是临时选择临时提取的文件夹,不用管直接ok:

提取完毕,程序会检查兼容性:

选择【同意并继续】,选择【自定义】安装:

在CUDA中取消勾选Visual Studio Integration,这是VS的插件,容易导致安装失败,我因为用pycharm所以就不勾选了

后面还有三个选项。NVIDIA GeForce Experience 是显卡驱动程序,如果你已经更新了显卡驱动版本,就不用选了,因为每个CUDA版本里面都含显卡驱动程序,如果你每次都选相当于每次更新显卡驱动版本。Other components 是声卡等设备驱动程序,我电脑有装了,不用选,最后一个也是。(大家可自己根据自身情况)

然后选择安装路径,点击安装,本地版的很快,在线版的需要联网下载安装,慢些。

等待安装完毕,可以检查是否安装成功,在终端键入命令nvcc -V,查看CUDA版本,能显示则成功了。

2.2 CUDNN安装

cuDNN下载好后解压,把里面的三个文件夹的内容分别移到CUDA安装目录里对应的同名文件夹里面。其中include和bin文件夹可以直接移动,lib文件夹内容要移动到lib\x64目录下

然后复制到CUDA的安装路径即可,我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6(注意!不是覆盖,是添加进去对应的文件夹里面去)

lib 是存在 lib/x64 里面

最后,配置下系统环境即可:
右键【此电脑】又名计算机—【属性】—【高级系统设置】-【环境变量】-下半部分【系统变量】里找到【Path】,双击打开(或者点编辑)

补全如下环境变量,如果你之前安装成功了CUDA,2、3条环境是系统已经自动添加了的。

2.3 TensorFlow-gpu 2.7安装

2.3.1 conda创建虚拟环境

  • 查看环境
conda info --env
  • 创建环境(例子为创建一个叫做tf27,python版本3.8的环境)
conda create -n tf27 python=3.8

2、删除环境(例子为删除名为tf27的环境)

conda remove -n tf27 --all

3、激活环境(例子为激活名为py36的环境)

activate tf27

4、退出环境

deactivate

5、从清华镜像源下载(例子为下载numpy)会显著提升下载速度

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

6、pip/anaconda直接修改镜像源,不用每次在后面加链接(Windows)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

后面直接安装即可

pip install tensorflow-gpu==2.7.0

其次,进入pycharm,创建.py文件进行测试:

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())

![在这里插入图片描述](https://img-blog.csdnimg.cn/554ef028f3fc4f57b8d50118312ff8e6.png#pic_center

输出版本号,和True

安装成功!

三、名词解释

TensorFlow:一个开源软件库,用于各种感知和语言理解任务的机器学习。

Keras:一个用Python编写的开源神经网络库,能够在TensorFlow、Microsoft Cognitive Toolkit、Theano或PlaidML之上运行。Keras旨在快速实现深度神经网络,专注于用户友好、模块化和可扩展性。

CUDA:统一计算架构。专为GPU同时处理多重任务而设计,大规模的并行计算处理,十分适合对图形处理、语音识别、视频等领域进行分析渲染。

cuDNN:深度神经网络库,针对CUDA优化,实现高性能GPU加速。

附录

网上查到的可行版本

python3.7.0+CUDA11.6.0+cuDNN8.3.2+tensorflow2.7.0+Keras2.7.0
python3.7.0+CUDA11.3.1+cuDNN8.2.1+tensorflow2.7.0+Keras2.7.0
python3.7.0+CUDA10.1.2+cuDNN7.6.5+tensorflow2.2.0+keras 2.3.1

1. 显卡算力和显卡驱动版本

显卡算力决定你的CUDA版本区间:
NIVDIA显卡算力官方查询

2.显卡驱动版本更新

显卡驱动版本,在NVIDIA控制面板的系统信息可查询到,建议更新到最新版本,这样就不会因驱动版本而下载CUDA版本受限。
NVIDIA官方显卡驱动更新

更新显卡驱动程序,建议在选项 Download Type 选 Studio Driver(SD),因为对深度学习有利。

NVIDIA官方 CUDA与显卡驱动版本对应表查询

3.tensorflow对应的CUDA、cuDNN和python版本

官方查询 tensorflow-gpu与CUDA cuDNN Python版本关系

参考

https://blog.csdn.net/typefree/article/details/117998734

https://blog.csdn.net/jiuzixu/article/details/122518914

tensorflow 安装GPU版本,CUDA与cuDNN版本对应关系,RTX3050Ti (notebook)相关推荐

  1. openEuler安装GPU、CUDA、cudnn

    openEuler安装GPU.CUDA.cudnn 准备工作 查看当前服务器显卡. 查看VGA接口显卡:lspci | grep VGA 查看Nvidia:lspci | grep NVIDIA [r ...

  2. 我的AI之路(5)--如何选择和正确安装跟Tensorflow版本对应的CUDA和cuDNN版本

    最新的Tensorflow和CUDA cuDNN的对应关系可以从这里找到: https://tensorflow.google.cn/install/source https://tensorflow ...

  3. [tensorflow]各个tensorflow版本和CUDA版本对应,以及各个GPU版本CUDA和cuDNN对应

    目录 各个CPU版本tensorflow对应的环境要求 各个GPU版本tensorflow对应的CUDA版本 各个版本的CUDA和英伟达显卡驱动对应表 缺失cudnn64_7.dll文件 查看本地CU ...

  4. tensorflow对应的python版本_详解Tensorflow不同版本要求与CUDA及CUDNN版本对应关系

    参考官网地址: Windows端:https://tensorflow.google.cn/install/source_windows CPU Version Python version Comp ...

  5. 干货:TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理

    最近发现很多QQ群和微信群里经常会有人问这么一个问题--"我安装TensorFlow GPU版本怎么总是提示CUDA版本不对或者cuDNN版本不对呢?"为了解决大家这个问题,我特意 ...

  6. tensorflow 各个版本的 CUDA 以及 Cudnn 版本对应关系

    本文由 简悦 SimpRead 转码, 原文地址 blog.csdn.net tensorflow 各个版本的 CUDA 以及 Cudnn 版本对应关系 一.tensorflow 各个版本需要的 CU ...

  7. Ubuntu 下 Pytorch, Tensorflow 对应的Python、英伟达显卡驱动、CUDA、CUDNN版本与环境信息查看方法

    人工神经网络已然成为现今最流行的机器学习工具,框架平台也在不断地完善升级,对GPU的依赖也越来越严重,在框架们升级的过程中会改进原有版本的错误并且提供很多美好的合理的新特性,会给使用者带来更多的便利, ...

  8. Win10多版本CUDA和CUDNN安装

    Win10多版本CUDA和CUDNN安装 一.CUDA安装 二.CUDNN安装 三.TensorRT安装 一.CUDA安装 点击CUDA安装包. 提取缓存文件. 检查系统兼容性,同意软件许可协议. 选 ...

  9. tensorflow和cuda以及cudnn版本对齐

    起因 今天想用下买了已经很久的GPU来玩tensorflow,毕竟显卡也不能只是用于玩游戏吧.然后发现因为cuda 和cudnn版本问题要卸载掉旧的重装新的,踩了一天的坑,来记录一下. cuda版本问 ...

  10. win10/11 paddlepaddle2.3/2.2 之 匹配CUDA和Cudnn版本安装

    完整的安装方法参考我的博客如下:包括显卡驱动检测 选择安装,VS2015/17/19安装,CUDA11.2安装和配置,cudnn安装.还包括CUDA卸载方法 (27条消息) 2022最详细,最新的 W ...

最新文章

  1. R语言sys方法:sys.info函数获取系统和用户信息、sys.localeConv函数获取当前区域中的数字和货币表示的详细信息、sys.setFileTime函数更改文件的时间
  2. C++找到一个大于或等于n且为2的幂的数字p的算法实现(附完整源码)
  3. 计算机启动和linux启动
  4. 考公专业科目计算机,2017国考中国证监会专业科目考试大纲(计算机类)
  5. 梨花带雨播放器程序开源版本
  6. 重庆自考学历计算机应用基础考试,2017年自考计算机应用基础模拟试题1
  7. 强大的Common类库
  8. IHS遥感图像融合算法及其相关的算法
  9. 七夕情人节在一起告白HTML源码(程序员专属情人节表白网站)
  10. dw常用标签_DW代码大全
  11. 【嵌入式工程师常用网站】
  12. 企业微信加好友太频繁怎么办?
  13. php内置函数分析之ucfirst()、lcfirst()
  14. 锁定计算机屏幕的快捷键是什么,锁住电脑屏幕的快捷键_打开电脑屏幕的快捷键...
  15. mysql获取今天的最后一秒_Oracle取得当天0时0分0秒和23时59分59秒
  16. 投资速度对天使投资人很重要——九合创投王啸
  17. 2019年全国大学生电子设计大赛(简单电路特性测试仪)
  18. 自定义Teams会议录像提示
  19. ThinkPHP5.1.x 框架源码分析之框架的灵魂
  20. mysql索引 b树_B树与MySQL数据库索引.ppt

热门文章

  1. 解决Visual Studio中scanf返回值被忽略问题
  2. LabelIng操作
  3. 语音识别基础(一):语音是什么
  4. ORACLE 11G RAC--CLUSTERWARE工具集1
  5. html5 假链接向左排列,最近很火的假红包强制加群引流方法
  6. Matlab--------股票接口,获取股票数据
  7. java文明用语的检测_阿里祭出大器,Java代码检查插件
  8. SOAP协议和restful 协议那些事儿
  9. 第5项修炼学习笔记:五项修炼系统思考强思维
  10. css transform组合