报错:tensorflow.python.framework.errors_impl.UnknownError: 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.
解决办法
ChatGPT一直和我说是CUDA和cuDNN的版本问题,但是上网搜索了TensorFlow与之对应的版本,发现没有问题。
在解决tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.中看到了两种解决办法,但是由于在 TensorFlow 2 中,tf.GPUOptions 和 tf.ConfigProto 已经被弃用,chatGPT推荐使用 tf.config.experimental 模块中的 API 进行 GPU 配置。
TensorFlow 2 中等价的代码是:
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
try:
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
except RuntimeError as e:
print(e)
亲测有效。