[问题] 设定gpu后仍占用另一个gpu(1000p)

楼主: znmkhxrw (QQ)   2018-10-15 17:55:00
这台电脑有2个GPU,概况如图

可是当我设定只吃 gpu = 0, fraction = 0.3 时
仍会吃掉 gpu = 1 的 fraction = 0.3 ,占著茅坑不拉屎

上图可以看到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