Problem:Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
Solve:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)or:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=args.gpu_memory_fraction,allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False))
Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
最新推荐文章于 2024-10-02 21:53:17 发布