当前位置:   article > 正文

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

tensorflow gpu版本安装

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的强大功能,我将在后续持续更新。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/95600
推荐阅读
相关标签
  

闽ICP备14008679号