Обучение моделей Keras на двух отдельных ноутбуках с юпитером на CPU и GPU - PullRequest
1 голос
/ 06 ноября 2019

Я тренирую модели Keras CNN для двух разных приложений на ноутбуке Jupyter. Учитывая, что я хочу использовать все ресурсы своего ПК, могу ли я использовать Keras-GPU в одном ноутбуке, а в другом - с использованием процессора.

Я узнал, что Keras использует GPU по умолчанию - если доступно - и я могу заставить Keras использовать CPU, как в Можно ли принудительно использовать Keras с бэкэндом Tensorflow для использования CPU или GPU по желанию? . Мой вопрос заключается в том, что при запуске этой строки кода



os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

изменятся настройки по умолчанию во всех работающих ноутбуках или только в этом конкретном ноутбуке?

1 Ответ

1 голос
/ 06 ноября 2019

запустив эту строку кода,

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

настройки по умолчанию меняются только в этом конкретном ноутбуке

Вы можете использовать

os.environ['CUDA_VISIBLE_DEVICES'] = ''

для тренировки на процессоре

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...