設置每個GPU內存佔用率,

tensorflow默認佔用全部gpu內存,這樣會出錯

設置每個GPU內存佔用率70%
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
sess = tf.Session(config=config)

ps:具體在一個程序中,只需將前兩句放到import一堆後面的位置,當全局變量

然後查找所有的tf.Session,加入config=config的參數

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