在 Tensorflow/Keras 中運行來自https://github.com/pierluigiferrari/ssd_keras的代碼時，使用估算器：ssd300_evaluation。我收到了這個錯誤。獲取卷積算法失敗。這可能是因為 cuDNN 初始化失敗，所以嘗試查看上面是否打印了警告日志消息。這與未解決的問題非常相似：Google Colab Error : Failed to get convolution algorithm.這可能是因為cuDNN未能初始化對于我正在運行的問題：蟒蛇：3.6.4。Tensorflow 版本：1.12.0。Keras 版本：2.2.4。CUDA：V10.0。cuDNN：V7.4.1.5。NVIDIA GeForce GTX 1080。我也跑了：import tensorflow as tfwith tf.device('/gpu:0'): a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a') b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b') c = tf.matmul(a, b)with tf.Session() as sess:print (sess.run(c))沒有錯誤或問題。
3 回答
暮色呼如
TA貢獻1853條經驗 獲得超9個贊
我遇到了同樣的問題，因此我解決了它：
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
或者
physical_devices = tf.config.experimental.list_physical_devices('GPU')
if len(physical_devices) > 0:
tf.config.experimental.set_memory_growth(physical_devices[0], True)
料青山看我應如是
TA貢獻1772條經驗 獲得超8個贊
我遇到了這個錯誤，我通過從我的系統中卸載所有 CUDA 和 cuDNN 版本來修復它。然后我安裝了CUDA Toolkit 9.0（沒有任何補?。┖?strong>cuDNN v7.4.1 for CUDA 9.0。
添加回答
舉報
0/150
提交
取消