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

news2024/12/27 15:06:30

安装步骤:

  • 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.coloradmin.cn/o/46301.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

【计算机】可信平台模块Trusted Platform Module - TPM

简述 Brief Introduction TPM内部功能模块示意图: 引述 Trusted Platform Module Technology Overview (Windows) | Microsoft Learn: Trusted Platform Module (TPM) technology is designed to provide hardware-based, security-related functions.…

速锐得解码理想汽车L8方向盘转向角度应用随动大灯照明升级

前日,速锐得解码了理想汽车L8车型,由于理想L8是新款车型,架构和理想L9十分相似,与理想ONE这一代有比较大的差异,这恰恰也是我们很好的一次学习机会,也让我们重新认识了理想汽车。 我这里,只挑有…

PCL 点云的法向量

一,点的法向量 点云法线 法向量的概念是很小的时候我们就已经说的,法向量是我们点云中一个非常重要的属性,诸如饿哦们常说的三维重建、点云分割,点云去噪 以及特种描述算法等。 特性: 点云中每一点的法向量夹角和曲率…

【即将开源】⽤于3D激光雷达SLAM闭环检测的词袋模型BoW3D

​以下内容来自从零开始机器人SLAM知识星球 每日更新内容 点击领取学习资料 → 机器人SLAM学习资料大礼包 #论文# BoW3D: Bag of Words for Real-time Loop Closing in 3D LiDAR SLAM 论文地址:https://arxiv.org/abs/2208.07473 作者单位:中国沈阳自动…

C++基础语法

cout输出 cin是键盘输入 //i input 输入 o output输出 stream流 输入输出流头文件&#xff08;类似stdio.h&#xff09; 2 #include <iostream> 3 4 //std(标准) 使用标准的命名空间using namespace std;//命名空间,此标识符作为此组群的名字 5 using namespace std; 6…

A*算法-Python实现

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

小程序数据请求的方式和注意事项

1.小程序中网络数据请求的限制 出于安全性方面的考虑&#xff0c;小程序官方对数据接口的请求做出了如下两个限制&#xff1a; ① 只能请求HTTPS类型的接口 ② 必须将接口的域名添加到信任列表中 2.配置request合法域名 假设要在自己的微信小程序中&#xff0c;希望请求某…

Mysql 索引基数与选择性

这篇文章主要介绍 MySQL 索引的 Cardinality 值&#xff08;基数&#xff09;以及索引的可选择性。 什么是索引&#xff1f; 先看一下 wiki 定义&#xff1a; 索引&#xff08;英语&#xff1a;Index&#xff09;&#xff0c;是一本书籍的重要组成部分&#xff0c;它把书中的…

微信小程序中基础入门

一、数据绑定 1.数据绑定的基本原则 ① 在data中定义数据&#xff08;在.js文件&#xff09; ② 在wxml中使用数据 2.Mustache语法的格式 把data中的数据绑定到页面中进行渲染&#xff0c;使用MUstache语法&#xff08;双大括号&#xff0c;可以理解为vue中的插值表达式&…

Redis——》数据类型:List(列表)

推荐链接&#xff1a; 总结——》【Java】 总结——》【Mysql】 总结——》【Redis】 总结——》【Spring】 总结——》【SpringBoot】 总结——》【MyBatis、MyBatis-Plus】 Redis——》数据类型&#xff1a;List&#xff08;列表&#xff09;一、简介…

Linux02——操作系统接口

一、前言 OS是软硬件之前的桥梁&#xff1a;操作系统管理硬件&#xff0c;最终以服务的形式提供给用户。如用户读取磁盘数据&#xff0c;OS设备管理将读出来的数据通过文件系统交给用户。OS管理员对CPU&#xff08;进程&#xff09;管理&#xff0c;对内存管理&#xff0c;对设…

PyCharm中鼠标悬停在函数上时显示函数和帮助

一、问题 1.1 鼠标放上去不显示文档的提示 鼠标放在随意一个函数上面不显示他的说明了 我也不知道是咋了 二、解决 2.1 首先我只记得有一个侧边栏叫document 经典的解决办法 2.2 在setting中查看这是干嘛的 很多东西都可以在setting中查看到具体的功能 还可以查看到从哪里能…

Redis高级篇——持久化

Redis持久化 1.RDB 1.1RDB简介 RDB全称Redis Database Backup file &#xff08;Redis数据备份文件&#xff09;&#xff0c;也被叫做Redis数据快照。把内存中的数据都记录到磁盘中&#xff0c;当Redis实例故障重启后&#xff0c;从磁盘中读取快照文件&#xff0c;恢复数据。…

蓝海创意云接受【看苏州】独家专访:助力苏州数字文化行业全方位发展

近日&#xff0c;由蓝海创意云提供渲染服务的动漫电影《老鹰抓小鸡》获金鸡奖最佳美术片提名&#xff0c;位列获奖名单的《长津湖》《独行月球》也由蓝海创意云渲染提供了后期服务。 就此&#xff0c;苏州广播电视总台旗下的苏州权威热点新闻和视频平台【看苏州】对蓝海彤翔执…

# Vue 中 JSON 编辑器使用

Vue 中 JSON 编辑器使用 文章目录Vue 中 JSON 编辑器使用背景描述vue-json-edit安装依赖测试页面效果图bin-code-editor安装依赖测试页面效果图CodeMirror安装依赖测试页面效果图vue-json-views安装依赖属性说明测试页面效果图vue-json-pretty安装依赖属性说明测试页面效果图码…

【CSS3】重点知识内容,快速上手

推荐前端学习路线如下&#xff1a; HTML、CSS、JavaScript、noodJS、组件库、JQuery、前端框架&#xff08;Vue、React&#xff09;、微信小程序和uniapp、TypeScript、webpack 和 vite、Vue 和 React 码源、NextJS、React Native、后端内容。。。。。。 层级选择器 a>b …

Kotlin高仿微信-第3篇-主页

Kotlin高仿微信-项目实践58篇详细讲解了各个功能点&#xff0c;包括&#xff1a;注册、登录、主页、单聊(文本、表情、语音、图片、小视频、视频通话、语音通话、红包、转账)、群聊、个人信息、朋友圈、支付服务、扫一扫、搜索好友、添加好友、开通VIP等众多功能。 Kotlin高仿…

Matlab图像处理基础(2):区域处理,边沿检测

目录 0. 概要 2. 卷积和相关 2.1 卷积 2.2 相关 2.3 卷积与相关的关系 2.4 Matlab函数 2.5 2-D卷积/相关的分解 3. 高通滤波&#xff0c;edge detection 3.1 matlab edge()函数 3.2 各种算子简介 3.2.1 Robert算子 3.2.2 Prewitt算子 3.2.3 Sobel算子 3.4 Cann…

维也纳国际酒店+丽柏酒店首个同物业双品牌项目成功加盟,中高端酒店品牌联动浮现新模式

从改革开放开启中国酒店市场化浪潮伊始&#xff0c;中国酒店市场一直处在高速发展的快车道。但四十年来&#xff0c;这条赛道上的竞争也出现了多次驱动引擎的动能转换。1996年起的20年里&#xff0c;庞大的经济型酒店切割了市场的主要蛋糕。而从2016年开始&#xff0c;中端酒店…

P2 Pytorch 张量数据类型

前言 1&#xff1a; 数据类型 2&#xff1a; 常用API 参考&#xff1a; 课时14 张量数据类型-1_哔哩哔哩_bilibili 一 数据类型 除了string ,相对于Numpy PyTorch 都能找到对应的数据类型 1.1 常用的Data type 常用的5种: IntTensor, LogTensor, ByteTensor, DoubleTensor,…