基于Tensorflow使用CPU而不用GPU問題的解決
更新時間：2020年02月07日 08:38:19 作者：JR_Chan
今天小編就為大家分享一篇基于Tensorflow使用CPU而不用GPU問題的解決，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
之前的文章講過用Tensorflow的object detection api訓(xùn)練MobileNetV2-SSDLite，然后發(fā)現(xiàn)訓(xùn)練的時候沒有利用到GPU，反而CPU占用率賊高（可能會有Could not dlopen library 'libcudart.so.10.0'之類的警告）。經(jīng)調(diào)查應(yīng)該是Tensorflow的GPU版本跟服務(wù)器所用的cuda及cudnn版本不匹配引起的。知道問題所在之后就好辦了。
檢查cuda和cudnn版本
首先查看cuda版本：
cat /usr/local/cuda/version.txt
以及cudnn版本：
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
重新安裝對應(yīng)版本Tensorflow
根據(jù)前面查看得到的cuda和cudnn版本，到Tensorflow官網(wǎng)查看對應(yīng)的Tensorflow-GPU版本，然后用conda install tensorflow-gpu=[version]重新安裝（把[version]換成對應(yīng)的版本比如1.12）就OK了。
以上這篇基于Tensorflow使用CPU而不用GPU問題的解決就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:
- 運(yùn)行tensorflow python程序,限制對GPU和CPU的占用操作
- 在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作
- tensorflow指定CPU與GPU運(yùn)算的方法實現(xiàn)
- 卸載tensorflow-cpu重裝tensorflow-gpu操作
- 使用Tensorflow-GPU禁用GPU設(shè)置(CPU與GPU速度對比)
- 詳解tf.device()指定tensorflow運(yùn)行的GPU或CPU設(shè)備實現(xiàn)
- 用gpu訓(xùn)練好的神經(jīng)網(wǎng)絡(luò),用tensorflow-cpu跑出錯的原因及解決方案
- Tensorflow中使用cpu和gpu有什么區(qū)別
- tensorflow之如何使用GPU而不是CPU問題
- TensorFlow安裝CPU版本和GPU版本的實現(xiàn)步驟
相關(guān)文章
Python守護(hù)進(jìn)程(daemon)代碼實例
這篇文章主要介紹了Python守護(hù)進(jìn)程(daemon)代碼實例,本文直接給出實現(xiàn)代碼,代碼中包含詳細(xì)注釋,需要的朋友可以參考下2015-03-03
Django 解決新建表刪除后無法重新創(chuàng)建等問題
這篇文章主要介紹了Django 解決新建表刪除后無法重新創(chuàng)建等問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了Python Pivot table透視表使用方法解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-09-09
解決Jupyter因卸載重裝導(dǎo)致的問題修復(fù)
這篇文章主要介紹了解決Jupyter因卸載重裝導(dǎo)致的問題修復(fù)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
偏函數(shù)(Partial function)是Python的functools模塊提供的一個很有用的功能,它允許我們通過固定部分參數(shù)或關(guān)鍵字參數(shù)來創(chuàng)建一個新的函數(shù),這篇文章主要給大家介紹了關(guān)于Python偏函數(shù)介紹及用法舉例詳解的相關(guān)資料,需要的朋友可以參考下2024-04-04
這篇文章主要介紹了Python中unittest用法,較為詳細(xì)的講述了unittest中相關(guān)函數(shù)的用法及完整實例,需要的朋友可以參考下2014-09-09
這篇文章主要為大家詳細(xì)介紹了Python如何通過WechatPCAPI來實現(xiàn)微信自動回復(fù)的功能，文中的示例代碼講解詳細(xì)，快跟隨小編一起動手嘗試一下2022-06-06
Pandas刪除數(shù)據(jù)的幾種情況(小結(jié))
這篇文章主要介紹了Pandas刪除數(shù)據(jù)的幾種情況(小結(jié))，詳細(xì)的介紹了4種方式，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-06-06
Python+Tkinter實現(xiàn)經(jīng)典井字棋小游戲
Tkinter是內(nèi)置到Python安裝包中的,只要安裝好Python之后就能import?Tkinter,而且IDLE也是用Tkinter編寫而成的。本文將用Tkinter編寫經(jīng)典的井字棋小游戲，需要的可以參考一下2022-03-03
pygame學(xué)習(xí)筆記（1）：矩形、圓型畫圖實例
這篇文章主要介紹了pygame學(xué)習(xí)筆記（1）：矩形、圓型畫圖實例,本文講解了pygame窗口、窗口退出、pygame中的顏色、圓形、矩形及一個完整實例,需要的朋友可以參考下2015-04-04