Test

报错

E tensorflow/stream_executor/cuda/cuda_dnn.cc:385] could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

解决方案（虽然不知道为啥），同样的环境配置，服务器上可以运行，自己台式机不行，可能显存小的原因？。。。

修改显存分配方式

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)