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

Question

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

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

Answer 1

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

Примечание.бэкэнд-библиотек PyCUDA.