[size=16px]英偉達1660s，最先安裝的tensorflow-gpu2.1.0，發現tf.test_is_gpu_available()是True，然而一運行resnet18，CPU100%占用率，而GPU幾乎不動，各種搜索，找不到問題的原因。
這個問題真的困擾了我好久，之前以為是版本太高，從tensorflow-gpu2.10降到了2.0.0，還是不行，對應的cuda、cudnn版本也都是按照官方檔案對應的。
都快要放棄了，認真檢查錯誤提示：
Loaded runtime CuDNN library: 7.4.1 but source was compiled with: 7.6.0.
于是下載了7.6.0的cudnn，替換掉，又出現問題了：
could not create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
沒關系，繼續搜索，
網上解決辦法
config = tf.ConfigProto()
session = tf.Session(config=config)
也不行，這是tensorflow1.X的版本代碼，所以要加上tf.compat.v1，不然找不到ConfigProto模塊。
變成：
config = tf.compat.v1.ConfigProto()
session = tf.Session(config=config)
結果還不行。
最后發現還少一行代碼，如圖：
大家只關注紅框里的代碼即可，其余代碼都是搜索出來，但是試了不管用的。
希望對同樣問題的人，有所幫助。[/size]
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/63753.html
標籤：Google技術社區
上一篇：lingo o-1混合整數規劃