Tensorflow2.0GPU占用量设置

Tensorflow2.0以下，设置gpu占比

os.environ['CUDA_VISIBLE_DEVICES'] = “1” #1卡

config = tf.ConfigProto()

config.gpu_options.per_process_gpu_memory_fraction = 0.2

 

Tensorflow2.0及以上，设置gpu占用量

os.environ['CUDA_VISIBLE_DEVICES'] = “1”#1卡

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')

tf.config.experimental.set_virtual_device_configuration(gpus[0],[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])