GPU版本 PaddlePaddle Win11 安装填坑

1,698 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第10天,点击查看活动详情

GPU版本 PaddlePaddle Win11 安装填坑

www.paddlepaddle.org.cn/

360截图20220805115657709.jpg

理论上有 Windows下的PIP安装-使用文档-PaddlePaddle深度学习平台 安装指南,安装应该不成问题,但是很多细小的问题不注意就会导致翻车。当然,其他的pytorch安装也是如此。

1.驱动安装

这应该是最简单的一步了,各种安装办法,这里我使用360驱动安装。

微信截图_20220805120059.png

微信截图_20220805120748.png

2.PaddlePaddle-GPU包安装


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

3.CUDA安装

这应该是最困难的一步了,稍有不慎就会失败。一般来说最新的驱动对应最新的CUDA,但是安装时也要考虑PaddlePaddle包与CUDA适配情况,搭配最新即可。这里选择CUDA11.2

官网链接: developer.nvidia.com/cuda-toolki…,进入之后选择合适的版本,此处选择CUDA11.2,此外还要下载对应的cudnn。

cuda_11.2.0_460.89_win10.exe
cudnn-11.2-windows-x64-v8.1.1.33.zip
  • 安装CUDA 11.2时不要勾选CUDA中的Visual Studio Intergration
  • 解压缩cudnn到cuda对应目录

如果勾选了的话,cuda可能默认使用显卡驱动的,与PaddlePaddle包要求不一致,会导致验证安装失败。

微信截图_20220805121324.png

4.验证


(p2) C:\Users\Administrator>python
Python 3.9.12 (main, Apr  4 2022, 05:22:27) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import paddle
>>> paddle.utils.run_check()
Running verify PaddlePaddle program ...
W0805 11:45:31.728643 11536 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 11.5, Runtime API Version: 11.2
W0805 11:45:37.219604 11536 gpu_resources.cc:91] device: 0, cuDNN Version: 8.1.
PaddlePaddle works well on 1 GPU.
PaddlePaddle works well on 1 GPUs.
PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.