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

Jupyter notebook Tensorflow GPU Memory 释放

时间:2017-12-13 11:49:28      阅读:1266      评论:0      收藏:0      [点我收藏+]

标签:prot   cti   链接   pos   body   not   .net   pre   book   

Jupyter notebook 每次运行完tensorflow的程序,占着显存不释放。而又因为tensorflow是默认申请可使用的全部显存,就会使得后续程序难以运行。暂时还没有找到在jupyter notebook里面自动释放显存的方法,但是我们可以做的是通过指定config为使用的显存按需自动增长,这样可以避免大多数的问题。代码如下:

gpu_no = ‘0‘ # or ‘1‘
os.environ["CUDA_VISIBLE_DEVICES"] = gpu_no

# 定义TensorFlow配置
config = tf.ConfigProto()

# 配置GPU内存分配方式,按需增长,很关键
config.gpu_options.allow_growth = True

# 配置可使用的显存比例
config.gpu_options.per_process_gpu_memory_fraction = 0.1

# 在创建session的时候把config作为参数传进去
sess = tf.InteractiveSession(config = config)

参考链接: http://blog.csdn.net/leibaojiangjun1/article/details/53671257

Jupyter notebook Tensorflow GPU Memory 释放

标签:prot   cti   链接   pos   body   not   .net   pre   book   

原文地址:http://www.cnblogs.com/zhsuiy/p/8030651.html

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