為tensorflow/tensorboard指定GPU,can not create session問題,No dashboards are active for the current da...

為tensorflow指定GPU,原因是,默認(rèn)創(chuàng)建session時,會將所有顯存占滿,發(fā)現(xiàn)有人在用的時候,就會session不能創(chuàng)建而報錯。
首先nvidia-smi查看顯卡的編號,最左邊一列,看看哪個空的

1.在終端設(shè)置使用的GPU

如果用方法一,雖然方便,但有的時候還是需要指定其他的GPU,這時可以這樣,例如

CUDA_VISIBLE_DEVICES=2 python test.py

這樣就只會使用序號為2的GPU

2.在程序中指定使用的GPU

import os

os.environ["CUDA_VISIBLE_DEVICES"]=‘2’

3.No dashboards are active for the current data set. 解決tensorboard無法啟動(can not create session)和顯示問題(no dashboards)

CUDA_VISIBLE_DEVICES=3 tensorboard --host=1.1.1.1 --port=6666 --logdir="/data1/wangning/tutorials/tensorflowTUT/tf14_tensorboard/

host是你自己的主機ip,port是你可以指定tensorboard的端口號,logdir要記得寫絕對路徑,不然總出現(xiàn)no dashboard錯誤

這里僅做一下小記錄,請忽略,以防清華鏡像不穩(wěn)定。
linux,GPU, py2.7

pip install \
  -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
  https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp27-none-linux_x86_64.whl

config = tf.ConfigProto(allow_soft_placement=True)

最多占gpu資源的70%

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)

開始不會給tensorflow全部gpu資源 而是按需增加

config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 網(wǎng)址 下載與安裝 你可以使用我們提供的 Pip, Docker, Virtualenv, Anaconda 或 源...
    九七學(xué)姐閱讀 4,920評論 3 11
  • 1. 介紹 首先讓我們來看看TensorFlow! 但是在我們開始之前,我們先來看看Python API中的Ten...
    JasonJe閱讀 11,981評論 1 32
  • feisky云計算、虛擬化與Linux技術(shù)筆記posts - 1014, comments - 298, trac...
    不排版閱讀 4,354評論 0 5
  • 我是莫默,今年21歲。就像你想的那樣,我的名字承載了我媽對我的殷切盼望,她希望我不要像我爸那樣不善言辭。可我想...
    涼安mmz閱讀 158評論 0 0
  • 文:漫橋煙雨 圖:源于網(wǎng)絡(luò) 你行走在黑夜里 越發(fā)明凈透亮 如此輝煌 萬物為你吟唱 你高高在上 為自己閃光 而我只是...
    漫橋煙雨閱讀 298評論 5 6

友情鏈接更多精彩內(nèi)容