【机器学习】P13 CUDA、cuDNN、Tensorflow 安装

安装 Tensorflow

在使用TensorFlow之前,需要将TensorFlow连接到GPU上以加速运算;
安装 Tensorflow 的整体步骤遵循:安装 CUDA,安装 cuDNN,安装 tensorflow-gpu:

  • CUDA(Compute Unified Device Architecture) 是NVIDIA提供的并行计算平台;
  • cuDNN(CUDA Deep Neural Network library) 是一种加速深度学习的库。安装它们可以帮助 TensorFlow 与 GPU 进行通信并加速计算。

安装 CUDA

1. 查看 CUDA driver 版本:
打开 NVIDIA 控制面板 -> 系统信息(左下角) -> 组件 -> 3D设置栏目中第三个;

在这里插入图片描述

2. 下载 CUDA Toolkit:
对照上一步查看的你的 CUDA 版本的 CUDA Toolkit 版本号进行下载:
比如我的 CUDA 版本是 11.8.88,我选择的是 11.8 版本的 CUDA Toolkit
https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述

3. 按照默认步骤安装,安装后打开 命令提示符 输入下述代码查看是否安装成功

nvcc -V

在这里插入图片描述

4. 若上述检查失败,极大可能原因是没有将 系统环境变量 配置:
我的电脑右键 -> 属性 -> 高级系统设置 -> 环境变量 -> 双击Path -> 新建
内容为:(具体地址按照读者自己安装的 CUDA 地址为准)

  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvv

在这里插入图片描述

配置完成系统环境后重试命令:

nvcc -V

安装 cuDNN

1. 还是查看 CUDA 版本,不过这次无需要那么细,只要看清楚是 11.x.xx 还是 12.x.xx就好;

2. 下载 cuDNN zip压缩包https://developer.nvidia.com/rdp/cudnn-download

在这里插入图片描述

3. 自己选择位置解压缩压缩包,解压缩后应该包含三个文件夹和一个文件:
binincludelibLICENSE

在这里插入图片描述

4. 将三个文件夹内容进行移动:

  • 首先打开 bin 文件夹:
    在这里插入图片描述
    复制全部文件,粘贴到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin 文件夹中;

  • 然后打开 include 文件夹:
    在这里插入图片描述
    同样复制全部文件,粘贴到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include 文件夹中;

  • 最后打开 lib\x64 文件夹:
    在这里插入图片描述
    将全部文件粘贴到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64 文件夹中;

5. 配置环境变量
cuDNN 中不会默认配置环境变量,需要读者手动配置环境变量:
首先打开 环境变量 path 栏目:

在这里插入图片描述

将解压缩后的 cuDNN 文件夹的 \bin 绝对地址复制粘贴:
D:\Program Files (x86)\cudnn-windows-x86_64-8.8.1.3_cuda11-archive\cudnn-windows-x86_64-8.8.1.3_cuda11-archive\bin

在这里插入图片描述

完成!

cuDNN的完成检查需要等待下一步 tensorflow-gpu 安装完成后;


安装 tensorflow-gpu

建议在 Anaconda Prompt 中创建一个新的环境,因为我的 base 环境已经安装好了 gpu 版本的 torch,而且如果都放在一个环境中会比较耗时;

1. Anaconda Prompt 中创建一个新的环境:

在这里插入图片描述

创建环境 tensorflow

conda create -n tensorflow pip python=3.6

激活环境 tensorflow

activate tensorflow

2. 查看下载 tensorflow-gpu 的版本号:

查看下载版本网址:https://www.tensorflow.org/install/source_windows

我的 CUDA 版本是 11.8.88,所以选择的 tensorflow-gpu 版本为 2.4.0

在这里插入图片描述

3. 下载 tensorflow-gpu:

打开 Anaconda Prompt:
输入命令:

pip install --ignore-installed --upgrade tensorflow_gpu==2.4.0
# 注意将2.4.0替换为你的版本号

等待安装完成!

4. Pycharm 中测试 tensorflow 成功调用 gpu:

import tensorflow as tf

# 检查是否有可用的GPU设备
if tf.config.list_physical_devices('GPU'):
    print('GPU可用')
else:
    print("GPU不可用")

在这里插入图片描述

5. 测试 cuDNN 可用:

import tensorflow as tf

# 检查TensorFlow-gpu是否可用
print("TensorFlow-gpu available:", tf.test.is_gpu_available())

# 检查cuDNN是否可用
print("cuDNN version:", tf.config.list_physical_devices('GPU'))

在这里插入图片描述

成功!!!

恭喜,撒花!!!


常见问题

1. Pycharm 中无法添加新的环境

Pycharm 中无法添加新的环境,比如刚才创建的 tensorflow,怎么办?

在这里插入图片描述
解决方案:
D:\Users\xhong\anaconda3\condabin\conda.bat 作为 Conda可执行文件,然后点击加载环境:

在这里插入图片描述

发现我们可以将刚创建的 tensorflow 添加到 pycharm 环境中了!

猜你喜欢

转载自blog.csdn.net/weixin_43098506/article/details/129927665
今日推荐