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

限制tensorflow训练时的显存设置,以及可见显卡号

时间:2020-01-04 12:32:08      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:device   mem   ssi   pre   训练   ble   memory   session   prot   

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = 0 #use GPU with ID=0
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # maximun alloc gpu50% of MEM
config.gpu_options.allow_growth = True #allocate dynamically
sess = tf.Session(config = config)

限制tensorflow训练时的显存设置,以及可见显卡号

标签:device   mem   ssi   pre   训练   ble   memory   session   prot   

原文地址:https://www.cnblogs.com/ywheunji/p/12148434.html

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