Слишком много ресурсов используется с Pytorch

Question

Я использую Pytorch для обучения модели DQN. В Ubuntu, если я использую htop, я получаю

Как видите, все ресурсы используются, и я немного волнуюсь по этому поводу. Вот мой код .

Есть ли способ использовать меньше ресурсов? Должен ли я добавить свое требование, используя pytorch?

Помните, что на моей машине нет графических процессоров, только процессоры

Answer 1

Да, есть. Вы можете использовать torch.set_num_threads(...) для указания количества потоков. В зависимости от версии PyTorch, которую вы используете, возможно, эта функция не будет работать правильно. Узнайте, почему в этом выпуске . Там вы увидите, что при необходимости вы можете использовать переменные окружения для ограничения использования потоков OpenMP или MKL с помощью OMP_NUM_THREADS=? и MKL_NUM_THREADS=? соответственно, где ? - количество потоков.

Сохранить в Имейте в виду, что эти вещи должны работать на графических процессорах с тысячами ядер, поэтому я бы ограничил использование ЦП только в случае крайней необходимости.