0. 环境
笔记本win10
NVIDIA GeForce GTX 1660 Ti
1. 安装cuda tool kit
paddlepaddle支持cuda 11.2 11.1 11.0 10.2,提示非安培架构的GPU,推荐使用CUDA10.2
https://developer.nvidia.com/cuda-toolkit-archive
选择10.2下载并安装
确认显卡驱动版本:
到这页面查看cuda driver要求的显卡驱动版本>=441.22:https://docs.nvidia.com/cuda/archive/10.2/cuda-toolkit-release-notes/index.html
右键桌面 -> NVIDIA控制面板 -> 系统信息 -> 看到驱动程序版本是472.47满足要求
安装cuda_10.2.89_441.22_win10.exe
安装cuda_10.2.1_win10.exe
安装cuda_10.2.2_win10.exe
2. cudnn
https://developer.nvidia.com/rdp/cudnn-archive
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows
2.1 zlib
下载并解压到D盘:http://www.winimage.com/zLibDll/zlib123dllx64.zip
Add the directory path of zlibwapi.dll to the environment variable PATH.
右键我的电脑 -> 高级系统设置 -> 环境变量 -> 编辑Path -> 添加D:\Programs\zlib123dllx64\dll_x64
2.2 安装
把
\cuda\cudnn-10.2-windows10-x64-v7.6.5.32\cuda
复制到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2添加到PATH
右键我的电脑 -> 高级系统设置 -> 环境变量 -> 编辑Path -> 添加C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
3. paddlepaddle
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html
目前飞桨支持的环境
Windows 7/8/10 专业版/企业版 (64bit)
GPU版本支持CUDA 10.1/10.2/11.0/11.1/11.2,且仅支持单卡
Python 版本 3.6+/3.7+/3.8+/3.9+ (64 bit)
pip 版本 20.2.2或更高版本 (64 bit)
3.1 python3.9
https://www.python.org/downloads/windows/
执行完安装包后添加环境变量D:\Programs\Python\Python39
3.2 再次确认cuda + cudnn版本
CUDA 工具包10.1/10.2 配合 cuDNN v7.6.5
3.3 CUDA10.2的PaddlePaddle
注意baidu的pypi境外无法访问。
python -m pip install paddlepaddle-gpu==2.2.2 -i https://mirror.baidu.com/pypi/simple
4. 验证
python 进入python解释器,输入
import paddle ,再输入
paddle.utils.run_check()