[問題] 設定gpu後仍占用另一個gpu(1000p)

作者: znmkhxrw (QQ)   2018-10-15 17:55:00
這台電腦有2個GPU,概況如圖 https://imgur.com/hkXJcsl
可是當我設定只吃 gpu = 0, fraction = 0.3 時
仍會吃掉 gpu = 1 的 fraction = 0.3 ,占著茅坑不拉屎 https://imgur.com/wsN6jI0
上圖可以看到gpu =0,1 都吃掉0.3, runnung process確實有兩個, 而只有0的有在動
而這台電腦重灌前確實是不會占著茅坑不拉屎的...
以下我附上嘗試過的指令、參考網站與環境,第一個幫解決的板友1000p奉上~感恩!
作者: eri820503 (snoopyahoo)   2018-10-15 20:56:00
import tensorflow as tfgpu_options=GPUOptions(per_process_gpu_memory_fraction=0.3)sess=tf.Session(config=tf.ConfigProto(gpu_tions=gpu_options))tf.keras.backend.set_session(sess)2跟3同一行, 4跟5同一行我這樣設定可以正常限制記憶體https://goo.gl/Cn2uak 可以參考這個如果是0吃30% 1吃0%,一開始設定os.environ就好但如果是0吃30% 1吃10%這種,可能要再找找之前用keras.backend去設定記憶體時,也有遇過設定了os.environ,但也會像這樣占用兩個GPU改用tf.keras.backend則不會

Links booklink

Contact Us: admin [ a t ] ucptt.com