tensorflowGPU显存占用

原创 2018年04月15日 19:43:36

问题

在毕设使用tensorflow在服务器上跑实验的时候遇到几个问题:

  • tensorflow默认占用所有GPU资源,因此启动就把所有GPU显存给占满。

解决办法:使用启用最少的GPU显存来运行程序或者限定显存比例

config = tf.ConfigProto() 
config.gpu_options.allow_growth = True 
session = tf.Session(config=config) 

config = tf.ConfigProto() 
# 占用GPU40%的显存,指定比例
config.gpu_options.per_process_gpu_memory_fraction = 0.4 
session = tf.Session(config=config)

另外一个问题是虽然指定了某个GPU运算,但还是会占用另外一个GPU的显存:

这里写图片描述
+ 因为在tensorflow代码中with tf.device(‘/gpu:0’):只会指定在GPU 0上计算,但仍然会默认占用所有GPU资源。

解决办法:添加如下代码:

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" //使用GPU0,这样就不会占用GPU1的显存

最终效果:
这里写图片描述

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_24306353/article/details/79952266

如何指定程序在某个GPU上运行

只需要添加如下两行代码:import os os.environ[“CUDA_VISIBLE_DEVICES”] = “1”注意,服务器看到的GPU序号和手动在Linux上查看的序号是相反的,比如 ...
  • qq_21460525
  • qq_21460525
  • 2017-06-19 21:25:02
  • 493

如何在基于tensorflow的深度学习框架keras中指定GPU

import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0,1" ...
  • weixin_38834204
  • weixin_38834204
  • 2017-05-18 10:36:57
  • 1290

Tensorflow 多GPU下,指定使用某一块GPU

如果机器中有多块GPU,tensorflow会默认吃掉所有能用的显存, 如果实验室多人公用一台服务器,希望指定使用特定某块GPU。 可以在文件开头加入如下代码:import os os.envir...
  • noirblack
  • noirblack
  • 2017-04-26 11:49:32
  • 7740

OS.ENVIRON()详解

我们想要用python获得一些有关系统的各种信息的时候就不得不想到os的environ,那这里面都具体包含了那些内容呢? 一、简介 对于官方的解释,environ是一个字符串所...
  • maoyongfan3
  • maoyongfan3
  • 2015-04-15 16:01:30
  • 1761

Linux 无进程显存占用问题

在linux命令行中键入nvidia-smi,显示没有进程但是GPU显存却使用很多 在命令行中输入sudo fuser -v /dev/nvidia*会显示如下top不会显示的进程 这些是我之前用...
  • Lillian_cl
  • Lillian_cl
  • 2017-10-09 10:23:01
  • 2126

如何获取显卡的GPU占用率和显存占用情况

http://blog.csdn.net/paopaoc/article/details/9093125 在游戏的性能调优过程中,经常会需要获取CPU占用率、IO、显卡GPU占用率等...
  • pizi0475
  • pizi0475
  • 2015-04-15 11:16:10
  • 20936

多人使用多GPU系统指定GPU,防止冲突

可以采取在当前用户下设置 vim ~/.bashrc 加入你想要使用的gpu的编号如下所示: export CUDA_VISIBLE_DEVICES=1 Only device 1 will b...
  • csuzhaoqinghui
  • csuzhaoqinghui
  • 2017-03-13 21:10:27
  • 942

keras tensorboard的使用, 设置指定GPU及其内存, 强制只使用cpu

1.强制只使用cpu: import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152 os.env...
  • xiaojiajia007
  • xiaojiajia007
  • 2017-06-05 12:10:21
  • 6878

10大游戏显存占用率测试

请注意这里的游戏全是开的最高效果!~    近几年,显卡的发展速度可以说快的惊人,几乎隔几个月,NVIDIA和AMD就会推出性能更高的新品。很显然,CPU的发展速度早已经不能和显卡相提并论了,N...
  • pizi0475
  • pizi0475
  • 2015-04-15 11:20:19
  • 2956

比较好用的显存测试软件

  • 2010年08月21日 10:49
  • 644KB
  • 下载
收藏助手
不良信息举报
您举报文章:tensorflowGPU显存占用
举报原因:
原因补充:

(最多只允许输入30个字)