Установка использования графического процессора в программе обучения факельным машинам

Question

В настоящее время я использую Theano для машинного обучения, теперь я хотел попробовать факел.

В Теано есть возможность установить флаги для использования GPU памяти:

os.environ["THEANO_FLAGS"] = "mode=FAST_RUN, device=gpu,floatX=float32,lib.cnmem=0.9"

Так Theano использует заданную мощность графического процессора, то здесь 90%. Однако в Torch аналогичная сеть использует нагрузку около 30% GPU.

Есть ли способ установить более высокую нагрузку на графический процессор в факеле, подобный Theano?

Answer 1

Факел будет использовать столько памяти GPU, сколько потребуется на основе стандартного распределителя.

Объем использования факела памяти не обязательно должен быть предварительно указан как пример, который вы делаете в Theano.