Выберите GPU во время исполнения в Theano

Я тренирую нейронные сети с theano и lasagne на машине с 4 GPU. Мой .theanorc содержит следующие строки:

[global]
device = gpu0

Поэтому, когда в Python я выполняю import theano, я получаю Using gpu device 0: GRID K520

Что если после импорта theano я решил использовать, скажем, gpu1? Я хотел бы сделать это динамически, то есть без редактирования .theanorc это возможно? Или даже выбрать его во время выполнения?

10 голосов | спросил P. Camilleri 13 PM000000120000001731 2015, 12:00:17

1 ответ


0

EDIT: Теперь Theano основан на серверной части массива GPU, и следующий API больше не доступен.

Как упомянул @EelkeSpaak, вы не можете изменить устройство GPU после того, как theano был импортирован. Но если вы хотите выбрать его программно, это возможно без изменения переменных среды.

  1. Убедитесь, что вы не выбираете устройство в своем файле .theanorc. Так что ничего подобного:

    device=gpu

  2. перед вызовом import theano выберите устройство с графическим процессором следующим образом:

    import theano.sandbox.cuda
    theano.sandbox.cuda.use('gpu1')
    
    #Results in Using gpu device 1: Tesla K80
    
ответил gidim 25 AMpMon, 25 Apr 2016 00:27:28 +030027Monday 2016, 00:27:28

Похожие вопросы

Популярные теги

security × 330linux × 316macos × 2827 × 268performance × 244command-line × 241sql-server × 235joomla-3.x × 222java × 189c++ × 186windows × 180cisco × 168bash × 158c# × 142gmail × 139arduino-uno × 139javascript × 134ssh × 133seo × 132mysql × 132