GPU版本 PaddlePaddle Win11 安装填坑
理论上有 Windows下的PIP安装-使用文档-PaddlePaddle深度学习平台 安装指南,安装应该不成问题,但是很多细小的问题不注意就会导致翻车。当然,其他的pytorch安装也是如此。
1.驱动安装
这应该是最简单的一步了,各种安装办法,这里我使用360驱动安装。
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包要求不一致,会导致验证安装失败。
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.