在Windows上安装Paddle、Tensorflow、PyTorch之前,需要安装CUDA和cudnn。
-
检查计算机平台的显卡驱动版本。
在NVIDIA控制面板主页查看(例如,我的版本号是466.81):
-
核查NVIDIA显卡和CUDA的版本号关系
在官网上可以查询,从下图可以看到驱动版本号为466.81对应的CUDA版本为CUDA 11.3.1 Update 1以下,因此可以选择CUDA 11.2(这是由于后边需要安装Paddle,Paddle目前支持的CUDA最高为11.2)
-
下载后安装。
CUDA官网下载地址(需要注册,免费下载,中间最好选择自定义安装,避免重复安装显卡驱动) -
在官网下载CUDA对应的cudnn。
在官网右上角的搜索栏中搜“cudnn”,下载后解压,将解压文件夹(bin,include,lib)整体复制粘贴到CUDA安装目录(默认目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2,粘贴需要管理员权限) -
完成安装