報錯原因： 內存分配出現了問題
解決方案： 在代碼前加上如下代碼

import tensorflow as tf gpus = tf.config.list_physical_devices("GPU")if gpus:gpu0 = gpus[0] #如果有多個GPU，僅使用第0個GPUtf.config.experimental.set_memory_growth(gpu0, True) #設置GPU顯存用量按需使用tf.config.set_visible_devices([gpu0],"GPU")

總結

以上是生活随笔為你收集整理的Failed to get convolution algorithm.This is probably because cuDNN failed to initialize的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。