Tensorflow ju lejon të specifikoni memorien maksimale të pjesshme GPU të përdorur nga procesi:
import tensorflow as tf
import keras
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.2)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
keras.backend.set_session(sess)
Duke përdorur sa më sipër, kodi im përfundon duke konsumuar rreth 2203MB/7982MB RAM GPU. SHËNIM: 2203 MB është më shumë se 20% e 7982.
Tani kodi im përfundon duke ekzekutuar herë pas here në GPU të ndryshme. Dhe përqindja e përdorimit të gpu nuk funksionon mirë në to (pasi GPU-ja e ndryshme ka sasi të ndryshme RAM)
A ekziston mundësia për të specifikuar fraksionin maksimal të memories GPU që do të përdoret në terma absolute
, në vend të termave relative
?
Duke kërkuar për diçka si (per_process_gpu_memory_inmb
:
# Looking for something like `per_process_gpu_memory_inmb` option
gpu_options = tf.GPUOptions(per_process_gpu_memory_inmb=2203)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
keras.backend.set_session(sess)