Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

上文要求各位检查自己显卡,和操作系统。

CUDA到下文地址,默认是最新的10.2,注意自己显卡驱动版本选择合适的版本。

https://developer.nvidia.com/cuda-toolkit-archive

Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

下载后默认安装, 记住路径,后面会用

CUDNN到下文下载,需要注册账号。agree后显示如下图片,选合适自己CUDA版本的CUDNN。

https://developer.nvidia.com/rdp/cudnn-download

Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

下载后解压缩,重命名为cudnn放到cuda安装目录下:

Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

 cudnn下目录结构如图,压缩包解压后会有个CUDA目录,CUDA目录里面才是我们要的,复制重命名的时候看仔细。Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

然后添加环境变量,如图,记得置顶。(如果出现path路径字节超2048的错误,网搜解决办法。)

Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

Tensorflow 2.0 目标识别(二)——CUDA&CUDNN安装

cmd下输入nvcc -V查看安装是否通过,等下一章吧。