tensorflow中指定GPU及GPU視訊記憶體設定

2021-08-06 04:59:55 字數 1255 閱讀 8652

檢視機器上gpu情況

命令: nvidia-smi

功能:顯示機器上gpu的情況

命令: nvidia-smi -l

功能:定時更新顯示機器上gpu的情況

其中左上側有0、1、2、3的編號,表示gpu的編號,在後面指定gpu時需要使用這個編號。

在終端執行程式時指定gpu

cuda_visible_devices=1   python  your_file.py

這樣在跑你的網路之前,告訴程式只能看到1號gpu,其他的gpu它不可見

可用的形式如下:

cuda_visible_devices=1           only device 1 will be seen

cuda_visible_devices=0,1         devices 0 and 1 will be visible

cuda_visible_devices="0,1"       same as above, quotation marks are optional

cuda_visible_devices=0,2,3       devices 0, 2, 3 will be visible; device 1 is masked

cuda_visible_devices=""          no gpu will be visible

在python**中指定gpu

import os

os.environ["cuda_visible_devices"] = "0"

設定定量的gpu使用量

config = tf.configproto()

config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用gpu90%的視訊記憶體

session = tf.session(config=config)

設定最小的gpu使用量

config = tf.configproto()

config.gpu_options.allow_growth = true

session = tf.session(config=config)

easyui 選中指定行

var rowindex 1 定義全域性變數,選中行索引 function pagination true,分頁 pagesize 15,預設選擇的分頁是每頁5行資料 pagelist 15,20,25,30 可以選擇的分頁集合 rownumbers true,行數 onloadsuccess fu...

ztree選中指定節點

var treeobj fn.ztree.getztreeobj bookcatalogtreelist 樹形結構的元素的id值 var snodes treeobj.getselectednodes var a treeobj.getnodebyparam gid 10 選中的那個元素的gid 1...

自動選中指定GameObject

自動選中指定gameobject,並在inspector面板顯示其屬性,如果已經在project檢視搜尋框輸入搜尋關鍵字,清除關鍵字可以先呼叫editorguiutility.pingobject方法。object o assetdatabase.loadassetatpath path edito...