MXNet - Как предотвратить полное выделение памяти

Question

Есть ли способ предотвратить полное выделение памяти GPU для MXNet?Так что он распределяет только то, что ему нужно, а не всю память графического процессора.

Я хочу использовать другую модель в Tensorflow / Keras на том же графическом процессоре наряду с MXNet, и кажется, что вся память зарезервирована MXNet.

Answer 1

MXNet выделяет память при необходимости .Возможно, в вашей программе произошла утечка памяти, или Tensorflow пытается предварительно выделить память для всего графического процессора, что является поведением по умолчанию .Такое поведение настраивается с помощью tf.GPUOptions .См. Ссылки о том, как использовать эти параметры.

Надеюсь, это поможет, Vishaal