安装tensorflow的GPU版本(详细图文教程)--CUDA11.6的安装_tensorflow cuda-程序员宅基地

技术标签: tensorflow  python  机器学习  深度学习  人工智能  

TensorFlow简介

TensorFlow是什么

  • TensorFlow是深度学习领域使用最为广泛的一个Google的开源软件库(最初由Google brain team进行开发的内部库,由于它的易用性Google决定把它开源出来).

  • 采取数据流图,用于数值计算.

    节点——处理数据

    线——节点间的输入输出关系

    数据流图中的数据叫做tensor, 表示张量, 即N维数据, tensor在数据流图中流动表示计算的过程, 这也是tensorflow名字的由来.

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-PFbJnP45-1654697407695)(.\img\数据流图.gif)]

  • 支持多种平台,GPU、CPU、移动设备

  • tensorflow特性:

    • 高度的灵活性: 只要能把数据的计算表示成数据流图就可以使用tensorflow
    • 真正的可移植性: 比如CPU、GPU、移动设备等等
    • 产品和科研结合
      • tensorflow研究最初是用于科研的,其实科研和工程还有一定的距离,科研的代码需要进一步各种各样的优化才能真正的做到产品上去,但是对于tensorflow则没有这个问题,Google团队把tensorflow优化的已经比较好了,做研究的代码可以无缝的用到产品上
    • 自动求微分
    • 多语言支持
      • tensorflow除了python以外,还支持各种各样的语言,比如说c++、java、javascript、R语言等
    • 性能最优化
      • 在tensorflow刚刚出来的时候由于它运行的比较慢,很多深度学习库呢都会拿tensorflow来进行比较,然后来证明自己比tensorflow好多少倍,但是随着tensorflow一步一步的进行开发,这种情况一去不复返了,tensorflow现在应该是运行最快的一个库,对于分布式的tensorflow来说,它的加速比几乎是线性的

tensorflow版本变迁

tensorflow 2.0 架构

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-pnFMJlYW-1654697407696)(.\img\TensorFlow2.0架构.jpg)]

  • tensorflow2.0主要特性:
    • 使用tf.keras和eager mode(动态图模式)进行更简单的模型构建.
      • 使用tf.data加载数据
      • 使用tf.keras构建模型,也可使用premade estimator来验证模型
        • 使用tensorflow hub进行迁移学习
      • 使用eager mode运行和调试
      • 使用分发策略来进行分布式训练
      • 导出到SavedMode
      • 使用TensorFlow Serve、Tensorflow Lite、Tensorflow.js部署模型
    • 鲁棒的跨平台模型部署
      • TensorFlow服务
        • 直接通过HTTP/RESR或GRPC/协议缓冲区
      • TensorFlow Lite——可部署到Android、iOS和嵌入式系统上
      • TensorFlow.js——在JavaScript中部署
      • 其他语言
        • C、Java、Go、C#、Rust、Julia、R等
    • 强大的研究试验
      • Keras功能API和子类API、允许创建复杂的拓扑结构
      • 自定义训练逻辑、使用tf.GraddientTape和tf.custom_gradient进行更细粒度的控制
      • 底层API自始至终可以与高层结合使用、完全的可定制
      • 高级扩展:Ragged Tensor、Tensor2Tensor等
    • 清除不推荐使用的API和减少重复来简化API

安装过程

常用IDE安装

python3.9的安装

在官网可以下载python3.9并安装好。这里我就不介绍了,想详细了解的可以看看我的这篇文档哦:机器学习常用的环境和工具安装和使用介绍

Anaconda的安装

这里我也不重点介绍了,我之前也重点详细地写过相关文章↓
还是搞不懂Anaconda是什么?读这一篇文章就够了
Jupyter notebook/Pycharm调用Anaconda虚拟环境
有需要的可以自行查看哦!一定会对你有帮助的!!!

CUDA安装

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。 开发人员可以使用C语言来为CUDA架构编写程序,所编写出的程序可以在支持CUDA的处理器上以超高性能运行。CUDA3.0已经开始支持C++和FORTRAN。

我们可以安装CUDA来进行深度学习在NVIDIA的GPU显卡加速运算。

但是我们在安装CUDA之前,要先确认计算机上是否支持CUDA程序的NVIDIA显卡设备。
打开设备管理器,查看显示适配器是否有英伟达的显卡配置
在这里插入图片描述
可以看出我的是1660Ti的配置支持。
如果计算机上没有NVIDIA显卡,则无法安装CUDA程序。
CUDA的安装很简单,就分成三步:
1、cuda的软件安装
2、cvDNN的神经网络加速库安装
3、配置环境变量

cuda软件安装

我们进入cuda软件下载的界面:CUDA下载页
进入后选择CUDA Toolkit 11.6.0版本
在这里插入图片描述在这里插入图片描述
大约是2.4GB,如果嫌麻烦也没关系,我已经下载好了,大家可以根据我分享的百度网盘下载:
链接:https://pan.baidu.com/s/10aHSylaMn8aEGj062c0HLA
提取码:qjuz
安装包安装完毕后,打开安装软件:路径可自行更改
请添加图片描述
请添加图片描述
请添加图片描述

这里我们选择自定义安装。

请添加图片描述
在组件CUDA一栏中,取消勾选Visual Studio Integration(因为我们并没有使用Visual Stduio环境,即使勾选上了也会安装失败)
请添加图片描述
在Driver components一栏比较Display Driver的新版本和当前版本的信息。若当前版本高于新版本,则取消勾选Display Driver;若若当前版本低于新版本,则保留默认安装信息即可,否则电脑会死机或者卡顿,甚至可能蓝屏。!!!
请添加图片描述
在CUDA的安装路径这里,保持默认就好,默然安装在C盘,一定一定不要修改。(来自一个手贱的人的警告)
一定一定要记住安装路径,因为后面配置环境要用到!!!
在这里插入图片描述
请添加图片描述
请添加图片描述
请添加图片描述
安装完成后,我们打开环境变量查看环境是否配置好了,打开系统变量:
请添加图片描述
如果系统变量没有自动配置的话,需要我们手动配置。路径是根据前面是否自己有没有修改来指定。↓
请添加图片描述
配置好环境变量后,我们检查下CUDA是否安装成功。打开cmd,输入以下命令查看CUDA是否安装成功(二选一)
如果不能显示以下信息,则说明安装失败。

nvcc -V

或者

nvcc --version

请添加图片描述
还可以查看CUDA 设置的环境变量。

set cuda

请添加图片描述
我们还可以搜索CUDA 的安装目录,找到“nvcc.exe”文件。
请添加图片描述

cuDNN神经网络加速库安装

CUDA并不是实现GPU的神经网络加速库,如果希望针对的是神经网络进行加速,我们还需要安装cuDNN神经网络加速库。
cuDNN并非是应用程序,而是几个文件包,下载后把它复制到CUDA 的目录下即可。
cuDNN下载页:cuDNN下载页
请添加图片描述
请添加图片描述
注意!如果要下载cuDNN,必须要登录NVIDIA的账户。登录完成后才能下载。没登录过的先注册。 如果大家嫌麻烦的话也不要紧,我已经下载好了,大家下载我的百度网盘分享连接下载即可:
链接:https://pan.baidu.com/s/10aHSylaMn8aEGj062c0HLA
提取码:qjuz
请添加图片描述
下载好安装包后,我们解压可以看到有四个文件:
请添加图片描述
我们查看CUDA11.6的原文件:
请添加图片描述
将cuDNN的文件全部复制到该文件夹下,复制后的文件展示:(有重复的文件是正常的,覆盖掉就好)
请添加图片描述
cuDNN其实就是CUDA的一个补丁而已,专为深度学习运算进行优化的,然后我们再添加环境变量!继续往下走。

配置环境变量

我们打开环境变量,在系统变量的path路径下添加以下路径:(具体要根据自己的安装路径下做调整)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include

添加好后是这样的:
在这里插入图片描述
配置好环境后,我们需要验证环境变量是否配置成功:
打开cmd,我们进入到以下路径:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite

然后分别执行以下两个命令:.\bandwidthTest.exe
和.\deviceQuery.exe
请添加图片描述
请添加图片描述
如果Result都为PASS的话则配置成功!

都安装好之后,我们可以继续输入nvidia-smi查看CUDA的信息,然后根据安装版本的信息再去实现其他的库(环境)安装和使用!
在这里插入图片描述
如图所示,可以看到驱动的版本是511.23;最高支持的CUDA版本是11.6版本。

TensorFlow的gpu版本安装

国内利用pip命令下载安装经常会遇到下载速度很慢甚至连接断开、响应超时等导致安装失败的情况。这时,我们可以选择国内的镜像配置pip源,仅需要在“pip install”命令后加入“-i 源地址”即可。
现在我们利用国内清华源安装TensorFlow的最新版本。
打开cmd,以下命令安装:

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

“-U”参数指定如果已安装此包,则进行升级命令。请添加图片描述
安装好后,我们检测是否安装成功:
进入python环境,打开ipython交互命令终端,导包:import tensorflow as tf

若无错误信息,输入

tf.test.is_gpu_available()

会返回tensorflow的gpu版本信息。
在末尾如果显示True,则tensorflow的gpu版本安装成功;若为False,则说明安装失败,需要重新检查CUDA,cuDNN的安装及其环境变量的配置。注意看返回的错误信息,重点检查CUDA和cuDNN的版本和tensorflow的版本是否匹配!请添加图片描述
或者输入

tf.config.list_physical_devices('GPU')

还能查看服务类型。
请添加图片描述
还可以查看可用的gpu数量:

print('Num GPUs Available:',len(tf.config.list_physical_devices('GPU')))

请添加图片描述
还可以查看TensorFlow的版本信息:

tf.__version__

在这里插入图片描述

后面如果我们通过tensorflow进行模型训练,系统会自动使用GPU来训练,不用我们自己手动设置。

最后,希望这篇文章可以帮助到你!

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_56197703/article/details/125192385

智能推荐

编译 Hadoop Eclipse Plugin-程序员宅基地

文章浏览阅读921次。1.首先从 git 下载源码 下载地址: https://github.com/winghc/hadoop2x-eclipse-plugin百度网盘下载: 链接:http://pan.baidu.com/s/1o6DMY4U 密码:y95i下载 ant-1.9.6 下载地址:http://mirrors.noc.im/apache//ant/binaries/apache-ant-1

Github删除仓库_这个仓库好像已经被从文件系统中删除,或者被移动到了其他地方。 你要删除这个标签-程序员宅基地

文章浏览阅读642次。既然写到了github这一块的知识点,那么,简单的说一下怎么删除一个仓库(项目)1:创建一个仓库2:填写仓库的名称3:创建完成4:如果要删除这个仓库的话,进入settings拉到最底下的Danger Zone5:输入仓库名称,Repository name,我的仓库名称是protest,删除即可原文作者:祈澈姑娘 技术博..._这个仓库好像已经被从文件系统中删除,或者被移动到了其他地方。 你要删除这个标签

引用 android string.xml文件中的整型和string型代替_android string.xml中引用其他值-程序员宅基地

文章浏览阅读4k次。在android的开发中,经常会遇见一句话,比如“我今年23岁了”;这个23需要在程序中生成,但是遇到一个问题,这完整的一句话是一个TextView中的,而不是三个textView拼接成的,而且是引用的string.xml文件中写好的;使用字符串拼接也可以实现,但是特别麻烦;今天遇到一个好方法,在string.xml中,不仅可以设置文字的格式换行等,还可以设置类似于变量的文本格式;_android string.xml中引用其他值

Hook 入门_hook入门-程序员宅基地

文章浏览阅读415次。Hook 简介为什么使用hook类组件和函数组件的区别没有破坏性改动Hook 使用规则useStateuseEffect自定义 HookuseRefuseContext为什么使用hookHook 是 React 16.8 的新增特性,是一些可以让你在函数组件里“钩入” React state 及生命周期等特性的函数。类组件和函数组件的区别详细参考 How Are Function Components Different from Classes?, 感兴趣的同学可以直接看原文。区别类组_hook入门

cruiscontroller的理解和学习_cruisbr-x-程序员宅基地

文章浏览阅读492次。CruiseControl:简称CC,持续集成工具,主要提供了基于版本管理工具(如CVS)感知变化或每天定时的持续集成,并提供持续集成报告、Email、Jabber等等方式通知相关负责人,其要求是需要进行日构建的项目已编写好全自动的项目编译脚本(可基于Maven或Ant)。在这里以一个项目来简要的说说cc的使用,通常项目对于日构建的类型的需求分为两种:1、每天的定时自动集成。2、感知版_cruisbr-x

《图形编程技术学习》(八)Windows配置freeglut、MAC配置freeglut以及配置glut_windows freeglut-程序员宅基地

文章浏览阅读834次。一.Windows配置freeglutWindows:1.下载Freeglut源码:http://freeglut.sourceforge.net/,2.VS2015中编译该源码工程,生成.lib和.dll文件(.h文件就在源码中)。(你也可以下载最新的,但相应操作可能会有所不同)3.将三类文件拷贝到相应目录下就完成了安装。具体 而言:如果是32位版本,则:1.\inc..._windows freeglut

随便推点

3.2.2 Python之pandas详解1-Series_from pandas import series什么意思-程序员宅基地

文章浏览阅读432次。Pandas是基于NumPy的一个常用库。之所以如此,是因为不论是读取还是处理数据,用它都非常简单。1,pandas基本数据结构Pandas有两种自己独有的基本数据结构。pandas虽然有两种数据结构,因为他是Python的一个库,所以Python的数据类型在这里依然适用,同样还可以使用类自己定义数据类型。Pandas里面又定义了两种数据类型:Series和DataFrame基本的导入方式:fro..._from pandas import series什么意思

【UE4 shader】扰动水流Shader_ue扰动-程序员宅基地

文章浏览阅读1.1w次,点赞11次,收藏81次。关键词:Fresnel(Lerp)、噪波(Panner Rotate TexCoord Time)、半透明、反射(屏幕空间反射SSR)、折射(Lerp IOR)、边缘波浪(世界位置偏移)、水波法线 、颜色变化-------------------------------------------------------------------------------Fresnel:用Lerp_ue扰动

利用tensorflow训练自己的图片数据集——数据准备_tensorflow训练自己的图库-程序员宅基地

文章浏览阅读2.3w次,点赞15次,收藏172次。昨天实现了一个简单的CNN网络。用了MNIST数据集,虽然看来对这个数据集用的很多,但是真正这个数据集是怎么在训练的时候被调用的,以及怎么把它换成自己的数据集都是一脸懵。直接附上链接:MNIST数据集解析作者给的代码是python2.x版本的,我用的python3.5,改了一些错误。import numpy as npimport structimport osimport m..._tensorflow训练自己的图库

day15_常用API-程序员宅基地

文章浏览阅读51次。day15_常用APIStringString是 java.lang.String , 使用不需要导包String是字符串类,是引用类型,底层是char数组,所以String的特性几乎和数组一致1 字符串一旦创建,该字符串对象不能更改2 为了提高字符串的访问效率,java虚拟机使用了一种缓存技术,可以对字符串操作更加简单方便,更加高效(字符串常量池)3 字符串会被保存在静态区中的常量池中,可以复用性增强当我们用到一个重复的字符串的时候,会去常量池中进行检索,如果有该字符串,则直接指向,如果

日常运维管理技巧十(linux下抓包 tcpdump)_tcpdump -nni-程序员宅基地

文章浏览阅读1.4k次。十、linux下抓包tcpdump命令是一款抓包工具,它可以打印所有经过网络接口的数据包的头信息,也可以使用-w选项将数据包保存到文件中,方便以后分析。1、tcpdump工具需要自行安装:# yum install -y tcpdump格式:tcpdump 选项可用选项,常见的我用红色标出:-a:尝试将网络和广播地址转换成名称;-c<数据包数量>:收到指定的..._tcpdump -nni

win7免费升级到win10专业版(正,亲测有用)_win7离线升级win10安装包-程序员宅基地

文章浏览阅读1.2k次,点赞2次,收藏10次。工具: MediaCreationTool20H2.exe_win7离线升级win10安装包

推荐文章

热门文章

相关标签