how to set the usage of gpu

[lxztju]

In tensorflow, we could use these commends to set the usage of gpu
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存
session = tf.Session(config=config)
If I use the package of slim, how to set the usage of gpu?