欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

基于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)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • Python守護進程(daemon)代碼實例

    Python守護進程(daemon)代碼實例

    這篇文章主要介紹了Python守護進程(daemon)代碼實例,本文直接給出實現(xiàn)代碼,代碼中包含詳細(xì)注釋,需要的朋友可以參考下
    2015-03-03
  • Django 解決新建表刪除后無法重新創(chuàng)建等問題

    Django 解決新建表刪除后無法重新創(chuàng)建等問題

    這篇文章主要介紹了Django 解決新建表刪除后無法重新創(chuàng)建等問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-05-05
  • Python Pivot table透視表使用方法解析

    Python Pivot table透視表使用方法解析

    這篇文章主要介紹了Python Pivot table透視表使用方法解析,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
    2020-09-09
  • 解決Jupyter因卸載重裝導(dǎo)致的問題修復(fù)

    解決Jupyter因卸載重裝導(dǎo)致的問題修復(fù)

    這篇文章主要介紹了解決Jupyter因卸載重裝導(dǎo)致的問題修復(fù),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-04-04
  • Python偏函數(shù)介紹及用法舉例詳解

    Python偏函數(shù)介紹及用法舉例詳解

    偏函數(shù)(Partial function)是Python的functools模塊提供的一個很有用的功能,它允許我們通過固定部分參數(shù)或關(guān)鍵字參數(shù)來創(chuàng)建一個新的函數(shù),這篇文章主要給大家介紹了關(guān)于Python偏函數(shù)介紹及用法舉例詳解的相關(guān)資料,需要的朋友可以參考下
    2024-04-04
  • Python中unittest用法實例

    Python中unittest用法實例

    這篇文章主要介紹了Python中unittest用法,較為詳細(xì)的講述了unittest中相關(guān)函數(shù)的用法及完整實例,需要的朋友可以參考下
    2014-09-09
  • 基于Python實現(xiàn)微信自動回復(fù)功能

    基于Python實現(xiàn)微信自動回復(fù)功能

    這篇文章主要為大家詳細(xì)介紹了Python如何通過WechatPCAPI來實現(xiàn)微信自動回復(fù)的功能,文中的示例代碼講解詳細(xì),快跟隨小編一起動手嘗試一下
    2022-06-06
  • Pandas刪除數(shù)據(jù)的幾種情況(小結(jié))

    Pandas刪除數(shù)據(jù)的幾種情況(小結(jié))

    這篇文章主要介紹了Pandas刪除數(shù)據(jù)的幾種情況(小結(jié)),詳細(xì)的介紹了4種方式,具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2019-06-06
  • Python+Tkinter實現(xiàn)經(jīng)典井字棋小游戲

    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學(xué)習(xí)筆記(1):矩形、圓型畫圖實例,本文講解了pygame窗口、窗口退出、pygame中的顏色、圓形、矩形及一個完整實例,需要的朋友可以參考下
    2015-04-04

最新評論