前提环境 python3.7,在anaconda虚拟环境下进行
首先需要确认的是电脑显卡最高所能支持的CUDA版本,支持的CUDA版本低,则不能安装高版本的CUDA工具包和加速计算库cudnn。支持的CUDA版本高,能安装当前版本以下的所有版本CUDA工具包和加速计算库cudnn。查看电脑NVIDIA显卡当前驱动程序所支持的CUDA版本,在Windows桌面下鼠标右击,选中NVIDIA控制面板,点击打开控制面板左下角的系统信息,点击‘显示’右边的‘组件’,看到NVCUDA64.DLL属性信息 CUDA后面的 ‘11.3.70’ 即为NVIDIA显卡当前驱动程序所支持的最高CUDA版本。
通过查看https://developer.nvidia.com/rdp/cudnn-archive找到对应的CUDA版本和cuDNN版本
找版本方法我的NVIDIA CUDA版本9.1.84 因此我需要找≤9.1
#安装CUDA工具包
conda install cudatoolkit=9.0
###安装cudnn加速计算库
conda install cudnn=7.6.4
#如果您的机器安装的是CUDA9,请运行以下命令安装
python -m pip install paddlepaddle-gpu==1.8.4.post97 -i https://mirror.baidu.com/pypi/simple
通过以上方法就可以成功安装GPU