Какие настройки мне нужны для настройки скорости графического процессора для обучения нейронных сетей?

Question

Я работаю с видеокартами Nvidia и хочу убедиться, что я в полной мере использую их производительность (безопасным способом, без разгона).Какие настройки мне нужно выполнить, чтобы полностью использовать их возможности?

Единственный параметр, который я использовал для более быстрого запуска, это:

nvidia-smi -pm 1

Answer 1

Вы можете настроить, сколько графического процессора вы хотите пожертвовать.

config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.4 session = tf.Session(config=config, ...)

Например, в этом примере из официальной документации ваша программа будет использовать 40% общей памяти вашего графического процессора.

Для более подробной информации вы можете проверить документацию