溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

如何解決基于Tensorflow使用CPU而不用GPU的問題

發布時間:2021-07-20 10:30:24 來源:億速云 閱讀:675 作者:小新 欄目:開發技術

這篇文章主要為大家展示了“如何解決基于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的問題”這篇文章的所有內容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內容對大家有所幫助,如果還想學習更多知識,歡迎關注億速云行業資訊頻道!

向AI問一下細節

免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI

亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女