Я установил версию тензора на GPU на Ubuntu 14.04.Как запустить Tensorflow на CPU
Я нахожусь на сервере графического процессора, где shadoworflow может получить доступ к доступным графическим процессорам.
Я хочу запустить тензорный поток на процессорах.
Обычно я могу использовать env CUDA_VISIBLE_DEVICES=0 для запуска на GPU no. 0.
Как я могу выбрать между процессорами вместо этого?
Я не интересуетесь переписано мой код с with tf.device("/cpu:0"):
Вы можете применить device_count параметр в tf.Session:
config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)
Смотрите также Protobuf конфигурационный файл:
Вы можете также установите для переменной среды значение
CUDA_VISIBLE_DEVICES=""
без изменения исходного кода.
^^ Это правильный ответ. –
Это также сработало для меня. – VAndrei
Кто-то сказал, что запущенные нейронные сети на процессорах после фазы обучения так же эффективны, как и их запуск на графических процессорах - то есть только для тренировочной фразы действительно нужен графический процессор. Вы знаете, верно ли это? Благодаря! – Crashalot
Если приведенные выше ответы не работает, попробуйте либо:
os.environ['CUDA_VISIBLE_DEVICES'] = ''
или
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
Спасибо, это намного лучше, чем другие варианты. – user1098761
Кто-то сказал, работает нейронные сети на процессорах после фазы обучения является столь же эффективным, как работает их на чипах - - т. е. только учебная фраза действительно нуждается в графическом процессоре. Вы знаете, верно ли это? Благодаря! – Crashalot
это работает для меня. очень просто – Shaowu
Это не работает для меня (tf1.1). Решение fabrizioM делает. – user1735003