报错如下:
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
首先检查自己的cudnn是否安装成功,在一切安装设置都正确的情况下,就是显存不足导致的。
原因:
TF2.0对RTX20系列显卡支持不是很好,容易爆显存,所以设置成用多少占多少显存,而不是一次性全占了(默认)。
解决办法:加入以下代码片段
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
try: