1. 程式人生 > >tensorflow GPU使用問題

tensorflow GPU使用問題

linux下檢視GPU使用情況的命令

nvidia-smi

這裡寫圖片描述

幾個觀察引數
Fan:代表顯示卡轉速,以百分比顯示;
Temp:代表顯示卡溫度;
Perf:效能等級,p1~p12,數字越小,效能較好。
Pwr:usage/cap :當前功率/總功率。
Memory-Usage:顯示卡的視訊記憶體使用容量跟顯示卡總的大小
GPU-Util:顯示卡計算單元使用率。
ok,多說一句,顯示卡也分為兩部分,即顯示卡跟計算單元,類似與記憶體跟CPU。

tensorflow的顯示卡使用方式

1、直接使用

with tf.Session() as sess:

這種方式會把當前機器上所有的顯示卡的剩餘視訊記憶體基本都佔用,注意是機器上所有顯示卡的剩餘視訊記憶體。因此程式可能只需要一塊顯示卡,但是程式就是這麼霸道,我不用其他的顯示卡,或者我用不了那麼多顯示卡,但是我就是要佔用。

2、分配比例使用

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction=0.6
with tf.Session(config=config) as sess:

其中這種方式跟上面直接使用方式的差異就是,我不佔用所有的視訊記憶體了,例如這樣寫,我就佔有每塊顯示卡的60%。

3. 動態申請使用

config = tf.ConfigProto()
config.gpu_options.allow_growth=True
with tf.Session(config=config) as sess:

這種方式是動態申請視訊記憶體的,只會申請記憶體,不會釋放記憶體。而且如果別人的程式把剩餘顯示卡全部佔了,就會報錯。

以上三種方式,應根據場景來選擇。
第一種因為是全部佔用記憶體,因此只要模型的大小不超過視訊記憶體的大小,就不會產生視訊記憶體碎片,影響計算效能。可以說合適部署應用的配置。
第二種和第三種適合多人使用一臺伺服器的情況,但第二種存在浪費視訊記憶體的情況,第三種在一定程式上避免了視訊記憶體的浪費,但極容易出現程式由於申請不到記憶體導致崩潰的情況。

4 指定GPU

在有多塊GPU的伺服器上執行tensorflow的時候,如果使用python程式設計,則可指定GPU,程式碼如下:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

相關推薦

TF-搭建tensorflow-gpu GPU環境

mnist 安裝 python 6.5 space 開發 with include 導致 看完官方的例子,自己憑著記憶敲出來mnist卷積的例子。跑起來才發現,憑著CPU跑不是一般的慢啊! train過程要叠代20000次,跑了1個小時沒跑完。也是怪我機子太差了。 於是下決

TensorFlow-GPU環境配置之一——安裝Ubuntu雙系統

啟動 windows src 技術 截圖 建立 交換空間 windows系統 空間 本機已經安裝過Windows系統,準備安裝Ubuntu雙系統進行TensorFlow相關工作,需要在windows中將磁盤分出一定空間供Ubuntu使用 1.首先下載Ubuntu17.04版

centos7下安裝部署tensorflow GPU 版本

tensorflow gpu   nvidia-modprobe nvidia-smi系統環境:centos71. 安裝 Python 2.7 # yum -y install zlib-devel bzip2-devel openssl-devel ncurses-devel sqlite-dev

windows7安裝tensorflow-gpu開發環境

www. software 清華大學 jetbrains 清華 數據分析師 數據分析 ftw 數據科學 1、安裝anaconda anaconda想必大家都不陌生,由於網站登不上去,我找到了清華大學的一個開源鏡像,下載地址為http://mirrors.tuna.tsin

utuntu安裝tensorflow-gpu

不同 flow ren nbsp code eve 遇到 sample ber 之前一直在windows下使用tensorflow,對cpu和gpu的安裝都是比較順利,都是使用anaconda環境下,創建虛擬conda環境,這樣方便不同python版本的管理。 一直也

TensorFlow-GPU環境配置之四——配置和編譯TensorFlow

http mage body class 調用 技術 圖片 環境 進行 首先,使用configure進行配置 配置完成後,使用bazel編譯retrain命令,編譯命令中加入--config=cuda即為啟用GPU 編譯進行中。。。 編譯完成 編譯完成後,調用ret

Tensorflow-gpu在windows10上的安裝(anaconda)

pri 安裝目錄 blank mirrors gpu版本 1.3 rep 檢查 科學計算 文檔來源轉載: http://blog.csdn.net/u010099080/article/details/53418159 http://blog.nitishmutha.com

win 10 下tensorflow-gpu配置

cudnn .aspx drive vid AC 所有 .cn down AI 小白一枚,請多指教,謝謝。 實踐證明,win10 + tensorflow1.6 + cuda9.1 +cudnn8.0 + python3.6安裝不適配(也許是個 人原因) 因為我的電

windows 上面的tensorflow-GPU、cuda、cudnn 安裝

atm search 容易 oca api RM run correct 既然 安裝說明 平臺:目前可在Ubuntu、Mac OS、Windows上安裝 版本:提供gpu版本、cpu版本 安裝方式:pip方式、Anaconda方式 Tips: 在Windows上目前支持p

Tensorflow+gpu安裝

which 配置文件 .com 虛擬 版本 post -a ffi oot 1、方法很簡單:在安裝好anaconda的前提下,直接用命令conda install tensorflow-gpu==1.4.1即可,其中1.4.1是版本號,可修改。 2、conda安裝包如果沒有

win10 python3.5 tensorflow(gpu)安裝

com 技術分享 ima uda -s href spa cudnn ron 為了避免麻煩,安裝的時候全部默認路徑 我安裝的CUDA和CUDNN版本 tensorflow版本1.7 這裏還有點小問題,直接import tensorflow有個報錯,我百度了錯誤

ubuntu14.04_64位安裝tensorflow-gpu

family 電腦配置 str 驅動 water list 出現 切換 名單 電腦配置:GeForce GTX 1080 安裝GTX1080驅動 進入英偉達官網,下載GTX1080驅動,開始搜索,然後下載所需版本即可。我下載的是最新的384.130。 也可以在這裏下載。

【深度學習篇】--Windows 64下tensorflow-gpu安裝到應用

archive dev upgrade eight ins 自己的 技術 tro tar 一、前述 一直以為自己的筆記本不支持tensflow-gpu的運行,結果每次運行模型都要好久。偶然間一個想法,想試試自己的筆記本,結果竟然神奇的發現能用GPU。於是分享一下安裝步驟.

工作筆記-服務器下ubuntu+cuda9.0+cuDNN+tensorflow-gpu環境搭建

應該 .bashrc run imp $path 系列 per 退出 一個 記錄實習工作中學到的新東西 服務器下ubuntu+cuda9.0+cuDNN7.0+tensorflow-gpu環境搭建 1.登陸服務器 windows系統下用xshell或者MobaXterm,新

Win10+TensorFlow-gpu pip安裝

x64 bsp eterm org flow path tps .org image 中文官網安裝教程:https://www.tensorflow.org/install/install_windows#determine_how_to_install_tensorflo

環境配置(近期實測)——Ubuntu16.04+CUDA9.0+tensorflow-gpu填坑記

u盤 earch 實測 win 1.3 ted 等待 gef kernel 近幾年深度學習在物體檢測方面出現了許多基於不同框架的網絡模型,不同模型需要不同的版本的Python、TensorFlow、Keras、CUDA、cuDNN以及操作系統。不得不說,要把經典物體檢測網絡

Tensorflow-GPU環境安裝

環境 WIN10+tensorflow1.8.0+cuda_9.0+cudnn 安裝資源 Anaconada: 連結:https://pan.baidu.com/s/1LAIXiolBb_RCbYmcxVQoDA 密碼:knak CUDA: 連結:https:

Ubuntu 16.04 配置安裝 Tensorflow Gpu版本

Ubuntu 16.04 配置安裝 Tensorflow Gpu版本 1、安裝CUDA8.0 2、更改gcc版本 3、安裝深度學習庫cuDNN 4、安裝 Tensorflow Gpu enable python 2.7 版本

win10+pycharm+anaconda3+tensorflow-gpu安裝

1.安裝好cuda和cuDNN 2.檢視所匹配版本進行安裝tensorflow-gpu。如下: tensorflow-gpu v1.9.0 | cuda9.0 |  cuDNN7.1.4可行  | 備註:7.0.4/ 7.0.5/ 7.1.2

[step by step]利用docker搭建Tensorflow環境(tensorboard + tensorflow+gpu

前言 本篇文章搭建環境的作業系統是ubuntu14,windows搭建docker的方式與ubuntu有所區別,win的使用者可以點選原文中的參考連線進行docker的搭建。掛載tensorboard的方法是一樣的,可供大家參考 搭建docker 官方文件 https://do