原文:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

在训练的脚本开头添加以下代码,指定使用的GPU,然后进行动态分配内存,重新运行就可以。 当然,出现这种问题也存在cudnn版本问题,但是发现这种情况并不多,大多是上面这种。 ...

2020-05-14 08:08 0 2782 推荐指数:

查看详情

Failed to get convolution algorithm解决

蒸腾了两天,终于搞定了 是cudnn版本的问题 更新cudnn的时候,首先要删除/usr/local/cuda-10.0/targets/x86_64-linux/lib路径下所有之前cudnn版本的库,sudo rm libcudnn* 然后 将下载的新版本的cudnn拷贝到 ...

Wed Feb 26 00:32:00 CST 2020 0 1066
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM