Tensorflow的gpu训练配置

开始训练时,cpu占用率过高,训练比较慢,原因在于cuda与tensorflow版本不匹配,而且没有安装cuDNN

Tensorflow官网对应表如下

重新安装cuda与对应版本的tensorflow和cuDNN解决问题,训练时间大大缩短,10epoch训练完毕大约5min

运行test_on_single_image.py得到推理结果