【AI学习笔记】TensorFlow GPU版本的安装(超详细)

news/2024/4/24 15:41:13/文章来源:https://blog.csdn.net/weixin_45954198/article/details/128076082

安装步骤:

  • 1. 确认显卡是否支持CUDA
  • 2. 安装CUDA
  • 3. 安装cuDNN
    • 3.1 安装 cudnn
    • 3.2 将cudnn64_8.dll存放的位置加入Path环境变量
  • 4. 安装TensorFlow GPU版本
    • 4.1 在Anaconda建立TensorFlow GPU虚拟环境
    • 4.2 安装Tensorflow-gpu
    • 4.3 安装Keras
  • 总结

1. 确认显卡是否支持CUDA

在安装之前,首先我们要确认现有的或预定要采购的独立显卡是否支持CUDA,具体操作如下:

  1. 查看支持CUDA的显卡,如下图:
    具体可到网站 https://developer.nvidia.com/cuda-gpus 查询。

在这里插入图片描述

  1. 查看显卡
    打开任务管理器(Ctrl+Shift+Esc)——>“性能”——>点击“GPU”,即可查看显卡名称(下图右上角)

    有了显卡名称就可以参考上图进行对照,查看是否支持CUDA。
    在这里插入图片描述

2. 安装CUDA

当我们确认系统已经安装了支持CUDA的显卡之后,就可以安装CUDA了,具体步骤如下:

  1. 下载并安装CUDA
    NVIDIA官网下载CUDA:https://developer.nvidia.com/cuda-downloads
    在这里插入图片描述
    注:Installer Type 两种安装方式的区别:
    ·exe(network):下载时文件比较小,后续执行安装时再下载其余部分。
    ·exe(local):下载时完整下载,后续执行安装时就不需要下载了。

  2. 设置解压缩安装程序的暂存目录
    在这里插入图片描述
    等待
    在这里插入图片描述

  3. 同意协议并继续
    在这里插入图片描述

  4. 选择精简安装
    在这里插入图片描述

  5. 警告未安装 Visual Studio
    CUDA 可以用 Visual Studio 开发。该界面警告未安装 Visual Studio。
    由于我们后续使用Python来开发,所以不需要安装 Visual Studio。
    在这里插入图片描述

  6. 下载 CUDA 界面
    在这里插入图片描述

  7. 确认安装
    在这里插入图片描述

  8. 安装完成
    在这里插入图片描述

3. 安装cuDNN

然后就是安装cuDNN,具体如下:

3.1 安装 cudnn

  1. 下载 cuDNN
    NVIDIA官网下载cuDNN:https://developer.nvidia.com/cudnn 。
    在这里插入图片描述

  2. 加入会员
    下载cuDNN必须先成为加速计算机开发者计划的会员。
    在这里插入图片描述

  3. 进入下载页面
    加入会员并且登录后,进入下载页面。
    在这里插入图片描述

  4. 查看下载后的文件
    下载后的文件 是一个ZIP压缩文件,在Windows10中可直接解压并打开。
    在这里插入图片描述
    在这里插入图片描述
    将其改名为“cuda”
    在这里插入图片描述
    ·然后将cuda放置一个合适的位置,如果不知道怎么放可以参考步骤5

  5. 复制到其他目录
    这里我直接将cuda放置在了我的D盘中的一个目录下。
    在这里插入图片描述

  6. 查看cudnn64_8.dll
    在D:\CS\Code\Jupyter\cuda\bin中(即 …/cuda/bin目录中)可以看到cudnn64_8.dll。这是动态链接程序库,其他程序会通过此链接库来使用cuDNN的功能。
    在这里插入图片描述

3.2 将cudnn64_8.dll存放的位置加入Path环境变量

为了让 Windows 系统知道所安装 cuDNN 的目录,必须设置Path环境变量,这样其他程序才能通过这个设置来存取 cudnn64_8.dll

  1. 打开系统环境变量
    首先点击“设置”,然后进行以下操作:
    在这里插入图片描述

  2. 编辑环境变量中的用户变量
    在Path环境变量中加入“D:\CS\Code\Jupyter\cuda\bin”(即cudnn64_8.dll的安装目录)
    在这里插入图片描述

在这里插入图片描述

4. 安装TensorFlow GPU版本

4.1 在Anaconda建立TensorFlow GPU虚拟环境

  1. 重新启动 “命令提示符” 程序,并切换到工作目录。
    在这里插入图片描述

  2. 在 Anaconda 建立 TensorFlow GPU 虚拟环境
    ·创建命令:conda create --name tensorflow-gpu python=3.9 anaconda
    在这里插入图片描述
    在这里插入图片描述

    按 y 键之后,就会开始安装Anaconda 虚拟环境,并且安装各个软件包。安装完成后屏幕显示界面如下:
    在这里插入图片描述

命令说明
conda create建立虚拟环境
–name tensorflow-gpu虚拟环境的名称是tensorflow-gpu
python=3.9Python版本是3.9
anaconda加入此命令选项,建立虚拟环境时,也会同时安装其他Python软件包,例如 Jupyter Notebook、Numpy、SciPy、Matplotlib、Pandas,用于进行数据分析
  1. 启用TensorFlow GPU 虚拟环境
    ·启动 Anaconda 虚拟环境:activate tensorflow-gpu
    在这里插入图片描述

4.2 安装Tensorflow-gpu

接下来,在TensorFlow GPU 虚拟环境中安装TensorFlow GPU版本
·安装命令:pip install tensorflow-gpu
在这里插入图片描述

4.3 安装Keras

然后是安装Keras
·安装命令:pip install keras
在这里插入图片描述

总结

在本篇我们主要介绍了如何安装CUDA、cuDNN,建立了TensorFlow GPU虚拟环境,并且在虚拟环境下安装了TensorFlow GPU版本与Keras。

补充:
如果第4步:安装TensorFlow GPU版本不太理解,可以参考TensorFlow CPU版本在Anaconda 虚拟环境中的安装。
具体可参考:【AI学习笔记】TensorFlow 与 Keras的安装(Windows Anaconda 虚拟环境版)

对于在TensorFlow GPU虚拟环境中测试GPU的强大功能,我将在后续持续更新。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.luyixian.cn/news_show_38796.aspx

如若内容造成侵权/违法违规/事实不符,请联系dt猫网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

供应荧光染料AF532 活性酯,AF532-NHS,CAS:477876-64-5

一:产品描述 1、名称 AF5 532酯 AF532-NHS AF532 活性酯 Alexa Fluor 532 AF532 NHS ester 2、CAS编号:477876-64-5 3、分子式:C34H33N3O11S2 4、分子量:723.77 5、质量控制:95% 6、储存: -20…

Arduino开发实例-DIY酒精浓度检测计

DIY酒精浓度检测计 在本文中,将详细介绍如何创建一个简单的酒精检测器。 它可以在各种应用领域中使用。市场上有许多先进的酒精传感器,价格合理,但我们在这里使用一些基本的微控制器来制作这个项目,如 Arduino、LED、蜂鸣器和 MQ3 酒精传感器。 1、MQ-3传感器介绍 MQ-3传…

A*算法-Python实现

好久没有在CSDN上发文章了,快一年了吧。这两天重新登录了一下,不看不知道,一看吓一跳,没想到访问量快13万了。 之前写博客的时候,想着把一些有用的东西写下来,一方面是当做笔记了,免得以后忘记…

5G无线技术基础自学系列 | SU-MIMO原理

素材来源:《5G无线网络规划与优化》 一边学习一边整理内容,并与大家分享,侵权即删,谢谢支持! 附上汇总贴:5G无线技术基础自学系列 | 汇总_COCOgsta的博客-CSDN博客 通过多天线技术支持单用户在上下行数据…

Vue3框架中路由的使用和局部刷新的功能(第十一课)

使用vue-router的步骤:p第一步:创建路由需要映射的组件(打算显示的页面);p第二步:通过createRouter创建路由对象,并且传入routes和history模式;配置路由映射: 组件和路径映射关系的routes数组&a…

DFL3:软件版本的选择和安装详解

这本是一个简单的问题,但是对于新手而言,所有问题,总是说的越清楚越仔细越好。我之所以这么说,肯定是有人问了。所以我就专门开一篇文章来说一说,软件版本的异同,以及如何选择。针对不同的语言,…

mysql相关基础知识篇(五)

1.MySQL 事务的四大特性说一下? 原子性:事务作为一个整体被执行,包含在其中的对数据库的操作要么全部被执行,要么都不执行。一致性:指在事务开始之前和事务结束以后,数据不会被破坏,假如 A 账户…

(2)点云库处理学习——剔除点云值

1、主要参考 1.1参考地址 (1) 点云离群点剔除 — open3d python_Coding的叶子的博客-CSDN博客_离群点去除 (2) open3d之点云异常值去除(笔记5)_Satellite_H的博客-CSDN博客 (3)斯坦福经典兔子的点云数据下载地址 下载地址 Model : Bunny 1.2兔子…

Git 打patch (打补丁)的使用

patch 的使用 一般是diff ,apply ,format-patch,am 1 生成patch git diff > test.patch 这个是打补丁(test.patch自己取的名字,这个命令可以看出没有指定修改的问题所以默认把所有修改的文件都打patch了,同时还需要注意,这里是本地修改的没有执行add缓存的) 如果想指定某…

[附源码]计算机毕业设计SpringBoot高血压分析平台

项目运行 环境配置: Jdk1.8 Tomcat7.0 Mysql HBuilderX(Webstorm也行) Eclispe(IntelliJ IDEA,Eclispe,MyEclispe,Sts都支持)。 项目技术: SSM mybatis Maven Vue 等等组成,B/S模式 M…

时序特征提取工具

在选择了需要提取的特征,确定了时序数据特征提取数据集的长度并对先验知识建模之后,就需要利用工具搭建特征提取系统。科研机构围绕不同问题域搭建的开源时序数据特征提取工具已经不少,我们可以利用这些工具快速实现希望达成的算法效果。下面…

驱动——platform驱动总线三种匹配方式

三种platform驱动匹配方式代码案例以及现象 方式一:通过设置名字进行匹配 相关API简介: 1、platform_device的API ①分配对象 struct platform_device { const char *name;//用于进行匹配的名字 int id;//总线号 PLATFORM_DEVID_AUTO(自…

ARM cortex-A7核UART实验 收发数据

头文件: 1 #ifndef __UART4_H__ 2 #define __UART4_H__ 3 4 #include "../common/include/stm32mp1xx_rcc.h" 5 #include "..…

【Android App】获取照片里的位置信息及使用全球卫星导航系统(GNSS)获取位置实战(附源码和演示 超详细)

需要全部代码请点赞关注收藏后评论区留言私信~~~ 一、获取照片里的位置信息 手机拍摄的相片还保存着时间、地点、镜头参数等信息,这些信息由相片接口工具ExifInterface管理,它的常用方法说明如下: getLatLong:获取相片拍摄时候的…

【人工智能 机器学习 深度学习】基础选择题1~30题 练习

目录 一、1~10题1.1 题目1.2 答案二、11~20题2.1 题目2.2 答案三、21~30题3.1 题目3.2 答案写在前面:适用于对 人工智能&机器学习&深度学习 进行复习的同学,同时,也可以通过基础题目的练习,加深理解。 一、1~10题 均是先给出10道题目,而后给出 10道题目的答案。 …

Python用广义加性模型GAM进行时间序列分析

每当你发现一个与时间对应的趋势时,你就会看到一个时间序列。我们围绕广义加性模型GAM技术进行一些咨询,帮助客户解决独特的业务问题。研究金融市场表现和天气预报的事实上的选择,时间序列是最普遍的分析技术之一,因为它与时间有着…

关于TreeView的简单使用(Qt6.4.1)

前言 TreeView是在Qt6.3中加入的,弥补了Qt中无官方树图。笔者上手尝试了下,虽然有点麻烦,但官方也做了不少简化。 本次教程,笔者创建一个简单的示例,以帮助读者使用TreeView。 一、创建模型类 当前模型需要使用C定义…

婚纱预订小程序开发,商家线上展示平台

婚纱代表着纯洁与忠贞,也是爱情永恒的见证者,穿上洁白的婚纱嫁给自己心爱的人是每个女生的梦想,婚纱对于每一个女生来说都有着重要的意义,所以选择一件美丽且适合的婚纱非常重要,因此人们在选择婚纱时会花费很多的时间…

Web3中文|区块链游戏的成长之痛

来源 | cointelegraph 编译 | DaliiNFTnews.com 在过去十年中,手机游戏已成为互动娱乐产业的重要支柱,得益于智能手机的普及,来自世界各地的用户都成为了硬核游戏玩家。 现在,区块链技术的出现正在推动一种范式的转变&#xff…

KNN最近邻算法分析及实现(Python实现)

KNN最近邻算法分析及实现(代码附录后文)1 KNN算法简介2 KNN基本原理3 简单实现KNN分析代码附录(Python):呆,站住别跑,留个赞,给个关注嘛都看到这了Author: Nirvana Of Phoenixl Proverbs for yo…