开始没有更新显卡驱动,导致下载了TensorFlow1.2+CUDA8.0+cudnn5.0,GPU版一直跑步起来,导致原本GPU版三个小时可以跑完的程序,CPU跑了一两天,所以还是要用GPU版,安装折腾了一天多重于安装成功,都是踩过的坑,流过的泪。
1:查看GPU型号
1 显卡型号支持:https://developer.nvidia.com/cuda-gpus
以下是我显卡对应的CUDNN5.0
2 更新显卡驱动:https://www.geforce.com/drivers
2安装cuda9.0
安装cuda9.0,期间如果安装失败,需要卸载后重新安装
网址:https://developer.nvidia.com/cuda-toolkit-archive
3安装cudnn7.3
网址:https://developer.nvidia.com/rdp/cudnn-archive