Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try loo。。。。。
解决方法:指定gpu
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'
本文解决了在使用GPU进行深度学习运算时遇到的cuDNN初始化失败问题,通过指定GPU设备,成功避免了该错误,确保了深度学习任务的顺利进行。
解决方法:指定gpu
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

被折叠的 条评论
为什么被折叠?