Укажите графический процессор Keras без использования CUDA_VISIBLE_DEVICES

У меня есть система с двумя графическими процессорами, и я использую Keras с бэкэндом Tensorflow. Gpu:0 выделяется для PyCUDA, которая выполняет уникальную операцию, которая передается Keras и изменяется с каждым пакетом. Таким образом, я хотел бы запустить модель Keras на gpu:1, оставив gpu:0 выделенным для PyCUDA.

Есть какой-либо способ сделать это? Просматривая предыдущие темы, я нашел несколько устаревших решений.


keras tensorflow artificial-intelligence gpu pycuda
person liqiudilk    schedule 25.10.2018    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Поэтому я не думаю, что эта функция в настоящее время осмысленно реализована в Keras. Нашел обходной путь, который я рекомендую: вы просто создаете несколько процессов, используя библиотеку многопроцессорности Python по умолчанию.

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

person liqiudilk    schedule 26.10.2018