为tensorflow/tensorboard指定GPU,can not create session问题,No dashboards are active for the current da...

为tensorflow指定GPU,原因是,默认创建session时,会将所有显存占满,发现有人在用的时候,就会session不能创建而报错。
首先nvidia-smi查看显卡的编号,最左边一列,看看哪个空的

1.在终端设置使用的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要记得写绝对路径,不然总出现no dashboard错误

这里仅做一下小记录,请忽略,以防清华镜像不稳定。
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)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 网址 下载与安装 你可以使用我们提供的 Pip, Docker, Virtualenv, Anaconda 或 源...
    九七学姐阅读 4,925评论 3 11
  • 1. 介绍 首先让我们来看看TensorFlow! 但是在我们开始之前,我们先来看看Python API中的Ten...
    JasonJe阅读 11,982评论 1 32
  • feisky云计算、虚拟化与Linux技术笔记posts - 1014, comments - 298, trac...
    不排版阅读 4,357评论 0 5
  • 我是莫默,今年21岁。就像你想的那样,我的名字承载了我妈对我的殷切盼望,她希望我不要像我爸那样不善言辞。可我想...
    凉安mmz阅读 159评论 0 0
  • 文:漫桥烟雨 图:源于网络 你行走在黑夜里 越发明净透亮 如此辉煌 万物为你吟唱 你高高在上 为自己闪光 而我只是...
    漫桥烟雨阅读 298评论 5 6

友情链接更多精彩内容