配置 paddlepaddle-gpu¶
目前 paddlepaddle-gpu 只支持到 python3.10,如果使用 python3.11,pip 会无法匹配到包。
此处(附录-使用文档-PaddlePaddle深度学习平台)有GPU 及其对应架构和推荐 CUDA 版本的列表,按照对应版本安装对应的 paddlepaddle-gpu。
比如我笔记本的 GTX1660Ti 是 Turing 架构,可以使用 CUDA10 和 CUDA11,不过 paddlepaddle 推荐对于安培以前的架构使用 CUDA10.2,性能会更优,因此选择
「CUDA 工具包 10.2 配合 cuDNN v7.6.5」
首先安装对应的 paddlepaddle-gpu包:
Text Only | |
---|---|
然后安装 CUDA 和 cuDNN
CUDA Installation Guide for Microsoft Windows (nvidia.com)
CUDA Toolkit 12.1 Update 1 Downloads | NVIDIA Developer
这个网站下方 Archive of Previous CUDA Release 中有以前的版本
Installation Guide - NVIDIA Docs
cuDNN Download | NVIDIA Developer
一切同理
跟着文档走,下载 Zlib 配置环境变量添加 zlibwapi.dll
所在路径到 PATH
,
然后下载 cuDNN 解压并复制内容到 CUDA 安装路径下。
然后理论上就OK了