您好,登錄后才能下訂單哦!
這篇文章主要為大家展示了“如何解決基于Tensorflow使用CPU而不用GPU的問題”,內容簡而易懂,條理清晰,希望能夠幫助大家解決疑惑,下面讓小編帶領大家一起研究并學習一下“如何解決基于Tensorflow使用CPU而不用GPU的問題”這篇文章吧。
檢查cuda和cudnn版本
?首先查看cuda版本:
cat /usr/local/cuda/version.txt
?以及cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
重新安裝對應版本Tensorflow
?根據前面查看得到的cuda和cudnn版本,到Tensorflow官網查看對應的Tensorflow-GPU版本,然后用conda install tensorflow-gpu=[version]重新安裝(把[version]換成對應的版本比如1.12)就OK了。
以上是“如何解決基于Tensorflow使用CPU而不用GPU的問題”這篇文章的所有內容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內容對大家有所幫助,如果還想學習更多知識,歡迎關注億速云行業資訊頻道!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。