在tensorflow上找到一个很好的解决办法:
把config.gpu_options.per_process_gpu_memory_fraction = 1
改成config.gpu_options.per_process_gpu_memory_fraction = 0.4
问题解决~
could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED解决办法
最新推荐文章于 2021-08-25 10:46:38 发布
在tensorflow上找到一个很好的解决办法:
把config.gpu_options.per_process_gpu_memory_fraction = 1
改成config.gpu_options.per_process_gpu_memory_fraction = 0.4
问题解决~