问题描述:

在这里插入图片描述

原因:

该项目使用了TensorFlow-GPU,电脑必须安装CUDA和CUDNN,也必须对应TensorFlow-GPU的版本。

解决:

1、安装cuda_10.0.130_411.31_win10

我所使用的的TensorFlow-GPU版本是1.13.1,下载CUDA10.0版本的,由于NVIDIA官网比较慢。
我把安装包上传到了百度网盘链接:https://pan.baidu.com/s/1cFLBAkxfp3pMIOfU1AGtkg
提取码:8ghs

在这里插入图片描述

按默认安装就行,会默认安装到C盘。

在这里插入图片描述

环境配置:(在安装的时候会默认配置到环境变量中,所以其实我们不需要自己配置path)

在这里插入图片描述

在这里插入图片描述

在CMD中输入nvcc -V查看CUDN版本

在这里插入图片描述

2、安装cudnn-10.0-windows10-x64-v7.6.4.38

百度网盘链接:https://pan.baidu.com/s/1toGXNPjTX7sqycfL6fS3Zw
提取码:emce

将CUDNN的bin、include、lib解压到CUDA路径下

在这里插入图片描述

3、添加环境变量

在这里插入图片描述

CUDNN完成配置

4、重启电脑

在这里插入图片描述

5、重新打开项目运行

读取到了我的垃圾1050TI显卡,说明上面的安装已经成功了!

在这里插入图片描述

随便说一下我的TensorFlow-GPU版本是1.13.1

在这里插入图片描述