tensorflow显存不够使用如何解决(windows系统)

 

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)

 

 

From: tensorflow显存不够使用如何解决(windows系统)

你可能感兴趣的:(人工神经网络)