tf.ConfigProto() 和tensorflow的GPU配置

# 使用0, 2, 3三塊GPU
os.environ['CUDA_VISIBLE_DEVICES'] = '0, 2, 3'

#設置每個GPU應該拿出多少容量給進程使用,0.6表示60%
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.6)

config=tf.ConfigProto(
	gpu_options=gpu_options,
	log_device_placement=True,  # log_device_placement是否打印設備分配日誌
	allow_soft_placement=True)  # allow_soft_placement指定設備不存在的話,是否允許TF自動分配設備

# 使用allow_growth option,開始分配少量顯存,然後按需增加, 一般和tf.GPUOptions選擇一個使用
# config.gpu_options.allow_growth = True

session = tf.Session(config=config, ...)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章