上文要求各位检查自己显卡,和操作系统。
CUDA到下文地址,默认是最新的10.2,注意自己显卡驱动版本选择合适的版本。
https://developer.nvidia.com/cuda-toolkit-archive
下载后默认安装, 记住路径,后面会用
CUDNN到下文下载,需要注册账号。agree后显示如下图片,选合适自己CUDA版本的CUDNN。
https://developer.nvidia.com/rdp/cudnn-download
下载后解压缩,重命名为cudnn放到cuda安装目录下:
cudnn下目录结构如图,压缩包解压后会有个CUDA目录,CUDA目录里面才是我们要的,复制重命名的时候看仔细。