在服务器运行深度学习代码时报错:(0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
解决方法:参考1、https://blog.csdn.net/weixin_45617478/article/details/107884222
2、https://blog.csdn.net/qq_38835585/article/details/108321628
首先,需要查看服务器的cuda版本+tensorflow版本
1.查看cuda版本:
在Xshell中执行命令:cat /usr/local/cuda/version.txt
2.查看tensorflow版本:
在jupyter中执行命令:import tensorflow as tf
tf.__version__
之后,若是版本不匹配直接进行修改,内存问题则添加内存分配代码。