Caffe и Tensorflow на Dell 7559 с технологией NVIDIA Optimus

Я купил ноутбук Dell 7559 для глубокого изучения. Я установил Ubuntu 16.04 на него, но у меня возникли проблемы с загрузкой caffe и tenorflow. В ноутбуке использовалась технология Nvidia Optimus для переключения между процессорами GPU и CPU для экономии заряда аккумулятора. Я проверил биос, чтобы увидеть, могу ли я установить его для использования только GPU, но нет никакой возможности для этого. Использование шмеля или nvidia-prime тоже не сработало. Теперь у меня Ubuntu 16 с настольной средой mate, которая мешает получить черный экран, но не помогает с проблемой cuda. Я был в состоянии установить драйверы и CUDA, но когда я собираю Caffe и TensorFlow, они не могут сказать, что он не обнаружил GPU. И я не смог установить OpenGL. Я пытался использовать несколько версий драйверов NVIDIA, но это не помогло. Любая помощь будет отличной. Благодарю.

1 ответ

Я думаю, что Bumblebee может позволить вам запустить Caffe/Tensorflow в режиме GPU. В целом, он также позволяет запускать другие программы CUDA на ноутбуке с технологией Optimus.

Если вы правильно установили Bumblebee (учебное пособие: Bumblebee Wiki для Ubuntu), вы можете вызвать бинарный файл Caffe, ожидая optirun перед бинарным кафе Итак, это выглядит так:

optirun ../../caffe-master/build/tools/caffe train --solver=solver.prototxt

Это работает и для сервера NVidia DIGITS:

optirun ./digits-devserver

Кроме того, Bumblebee также работает на моем настольном ПК с двумя графическими процессорами (Intel HD 4600 + GTX 750 Ti). Дисплей на моем ПК управляется Intel HD 4600 через порт HDMI на материнской плате. NVidia GTX 750 Ti используется только для программ CUDA.

На самом деле, для моего настольного ПК, nvidia-prime (на самом деле он вызывается через программу командной строки prime-select) используется для выбора графического процессора, который управляет рабочим столом. У меня есть встроенный графический процессор, подключенный к дисплею через порт HDMI и графический процессор NVidia через DisplayPort. В настоящее время DisplayPort неактивен. Сигнал дисплея поступает с порта HDMI.

Насколько я понимаю, PRIME делает это, изменяя /etc/X11/Xorg.conf сделать либо интегрированный графический процессор Intel, либо графический процессор NVidia текущим адаптером дисплея, доступным для X. Я думаю, что настройки PRIME имеют смысл только тогда, когда оба графических процессора подключены к какому-либо дисплею, что означает, что не должно быть связи Optimus между двумя графическими процессорами, такими как в ноутбуке (или для ноутбука с мультиплексором, таким как Dell Precision M4600, Optimus отключен в BIOS).

Более подробную информацию о Display Mux и Optimus можно найти здесь: Использование драйвера NVIDIA с ноутбуками Optimus

Надеюсь это поможет!

Другие вопросы по тегам