報錯原因： 內(nèi)存分配出現(xiàn)了問題
解決方案： 在代碼前加上如下代碼

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) #設(shè)置GPU顯存用量按需使用tf.config.set_visible_devices([gpu0],"GPU")

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。