问题描述:
原因:
该项目使用了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
评论(0)
您还未登录,请登录后发表或查看评论