2020年5月14日
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
摘要： 在训练的脚本开头添加以下代码，指定使用的GPU，然后进行动态分配内存，重新运行就可以。 import os os.environ['CUDA_VISIBLE_DEVICES'] = '0' config = ConfigProto() config.gpu_options.allow_growth 阅读全文
posted @ 2020-05-14 08:08 apan008 阅读(3306) 评论(0) 推荐(1) 编辑