Я хочу увеличить использование памяти GPU - PullRequest
Новая
       25

Я хочу увеличить использование памяти GPU

0 голосов
/ 27 мая 2020

Я использую код Mask R-CNN. Использую 2 компа. Первый - компьютер с одним графическим процессором (Geforce RTX 2080 Ti), а второй - компьютер с несколькими графическими процессорами (Geforce GTX 1080 Ti * 6). Я хочу провести 7 экспериментов одновременно, используя каждый графический процессор.

Однако проблема в том, что использование памяти графического процессора сильно отличается, поэтому я не могу завершить sh эксперименты за короткое время. Графический процессор первого компьютера использует 10934 МБ из 11018 МБ. Первый графический процессор второго компьютера использует только 772 МБ из 11171 МБ. Другие графические процессоры второго компьютера действуют так же, как первый графический процессор второго компьютера. Следовательно, один и тот же эксперимент проводится на втором компьютере более 5 раз.

Я использую точно такой же код. Есть только одно различие между первым компьютером и вторым компьютером. Я использовал виртуальную среду anaconda на втором компьютере для установки tensorflow (1.14.0) и Keras (2.1.3). И я не использую виртуальный env на первом компьютере.

Я пробовал несколько кодов:

1. 

gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
    try:
        tf.config.experimental.set_memory_growth(gpus[0], True)
    except RuntimeError as e:
        print(e)

2. 

config_yj = tf.ConfigProto()
config_yj.gpu_options.per_process_gpu_memory_fraction = 0.9
session = tf.Session(config=config_yj)

3. 

gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
keras.backend.tensorflow_backend.set_session(sess)

Однако все они не работали.

Есть ли ограничивающая опция в виртуальной среде anaconda? Я думаю, что проблема связана с виртуальной средой, потому что единственная разница в среде.

Очень хочу решить эту проблему. Я надеюсь, что вы можете мне помочь. Это первый раз, когда я задаю вопрос о stackoverflow, так что простите за дискурсивность. Если есть дополнительная необходимая информация, сообщите мне.

Спасибо.

...