基于Tensorflow使用CPU而不用GPU問(wèn)題的解決

更新時(shí)間：2020年02月07日 08:38:19 作者：JR_Chan

今天小編就為大家分享一篇基于Tensorflow使用CPU而不用GPU問(wèn)題的解決，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

之前的文章講過(guò)用Tensorflow的object detection api訓(xùn)練MobileNetV2-SSDLite，然后發(fā)現(xiàn)訓(xùn)練的時(shí)候沒(méi)有利用到GPU，反而CPU占用率賊高（可能會(huì)有Could not dlopen library 'libcudart.so.10.0'之類的警告）。經(jīng)調(diào)查應(yīng)該是Tensorflow的GPU版本跟服務(wù)器所用的cuda及cudnn版本不匹配引起的。知道問(wèn)題所在之后就好辦了。

檢查cuda和cudnn版本

首先查看cuda版本：

cat /usr/local/cuda/version.txt

以及cudnn版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

重新安裝對(duì)應(yīng)版本Tensorflow

根據(jù)前面查看得到的cuda和cudnn版本，到Tensorflow官網(wǎng)查看對(duì)應(yīng)的Tensorflow-GPU版本，然后用conda install tensorflow-gpu=[version]重新安裝（把[version]換成對(duì)應(yīng)的版本比如1.12）就OK了。

以上這篇基于Tensorflow使用CPU而不用GPU問(wèn)題的解決就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python實(shí)現(xiàn)合成多張圖片到PDF格式
在日常生活中，經(jīng)常會(huì)遇到需要提交身份證正反面證明資料的情況，而且這些網(wǎng)站大部分只接受pdf格式，這時(shí)候我們就需要把身份證正反面兩張圖片合成為一個(gè)pdf文件。本文將為大家提供用Python實(shí)現(xiàn)這一要求的方法，需要的可以參考一下
2022-02-02
Django跨域請(qǐng)求CSRF的方法示例
這篇文章主要介紹了Django跨域請(qǐng)求CSRF的方法示例，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧
2018-11-11
解決ImportError: cannot import name ‘Imput
您遇到的ImportError: cannot import name ‘Imputer‘錯(cuò)誤提示表明您嘗試導(dǎo)入一個(gè)名為’Imputer’的模塊或類,但是該模塊或類無(wú)法找到,本文小編給大家介紹了如何解決這個(gè)問(wèn)題,需要的朋友可以參考下
2023-10-10
基于pandas中expand的作用詳解
今天小編就為大家分享一篇基于pandas中expand的作用詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-12-12
tensorflow 變長(zhǎng)序列存儲(chǔ)實(shí)例
今天小編就為大家分享一篇tensorflow 變長(zhǎng)序列存儲(chǔ)實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-01-01
使用Python對(duì)零售商品進(jìn)行數(shù)據(jù)分析
這篇文章主要為大家介紹了使用Python對(duì)零售商品進(jìn)行數(shù)據(jù)分析詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
2022-05-05
Python函數(shù)裝飾器常見(jiàn)使用方法實(shí)例詳解
這篇文章主要介紹了Python函數(shù)裝飾器常見(jiàn)使用方法,結(jié)合實(shí)例形式分析了Python函數(shù)裝飾器的概念、原理、用法及相關(guān)操作注意事項(xiàng),需要的朋友可以參考下
2019-03-03
詳解python3 + Scrapy爬蟲(chóng)學(xué)習(xí)之創(chuàng)建項(xiàng)目
這篇文章主要介紹了python3 Scrapy爬蟲(chóng)創(chuàng)建項(xiàng)目，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2019-04-04
python 實(shí)現(xiàn)簡(jiǎn)單的計(jì)算器(gui界面)
這篇文章主要介紹了python 如何實(shí)現(xiàn)簡(jiǎn)單的計(jì)算器，幫助大家更好的理解和使用python，感興趣的朋友可以了解下
2020-11-11
Python Datetime模塊和Calendar模塊用法實(shí)例分析
這篇文章主要介紹了Python Datetime模塊和Calendar模塊用法,結(jié)合實(shí)例形式分析了Python日期時(shí)間及日歷相關(guān)的Datetime模塊和Calendar模塊原理、用法及操作注意事項(xiàng),需要的朋友可以參考下
2019-04-04