码迷,mamicode.com
首页 > 其他好文 > 详细

TensorFlow设置GPU占用量

时间:2017-12-16 21:15:06      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:alt   imp   igp   使用   div   gpo   flow   就会   env   

默认开启Tensorflow的session之后,就会占用几乎所有的显存,进行如下设置即可:

指定GPU编号:

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

或者在脚本或者命令行中指定

export CUDA_VISIBLE_DEVICES=1

 

指定GPU使用的是按需分配的:

config = tf.ConfigProto()  

config.gpu_options.allow_growth=True  

sess = tf.Session(config=config)

TensorFlow设置GPU占用量

标签:alt   imp   igp   使用   div   gpo   flow   就会   env   

原文地址:http://www.cnblogs.com/rainsoul/p/8047603.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!