ICode9

精准搜索请尝试: 精确搜索
首页?>?其他分享> 文章详细

基于tensorflow训练模型的显存不足解决办法

2019-10-14 23:53:01??阅读:8??来源: 互联网

标签:显存?解决办法?gpu?tf?tensorflow?config?options


?

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #指定第一块GPU可用
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 程序最多只能占用指定gpu50%的显存
config.gpu_options.allow_growth = True #程序按需申请内存
sess = tf.Session(config = config)



标签:显存,解决办法,gpu,tf,tensorflow,config,options

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有