深度学习GPU支持平台安装

280 阅读1分钟

以下内容是在学习过程中的一些笔记,难免会有错误和纰漏的地方。如果造成任何困扰,很抱歉。

首先打开NVIDIA控制面板,检查自己显卡最高支持的CUDA版本

选择合适的版本进行下载

下载地址
CUDA Toolkit Archive | NVIDIA Developer
cuDNN Archive | NVIDIA Developer

首先进入第一个链接进行安装

选择Windows

本人电脑是Win11

选择合适的安装目录

跑进度条

此时检查系统兼容性

安装流程一步步即可,接下来是去安装cuDNN

cudnn-11.4-windows-x64-v8.2.4.15

首次安装需要注册,这里不多解释,安装完成以后,解压出来的文件夹需要放在这个目录

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

然后去新增加环境变量

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

然后去Paddle官网下载合适版本的GPU库

python -m pip install paddlepaddle-gpu==2.3.2.post116 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

最后一步,检查是否安装成功

import paddle.fluid
paddle.fluid.install_check.run_check()

成功后的输出结果

Running Verify Fluid Program ... 
W1023 22:01:58.082381 26072 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 7.5, Driver API Version: 11.6, Runtime API Version: 11.6
W1023 22:01:58.104391 26072 gpu_resources.cc:91] device: 0, cuDNN Version: 8.2.
Your Paddle Fluid works well on SINGLE GPU or CPU.
Your Paddle Fluid works well on MUTIPLE GPU or CPU.
Your Paddle Fluid is installed successfully! Let's start deep Learning with Paddle Fluid now

你学废了吗