使用nvidia optimus技术的戴尔7559上的Caffe和Tensorflow

时间:2016-07-19 02:57:45

标签: gpu tensorflow nvidia caffe optimus

我买了戴尔7559笔记本电脑进行深度学习。我安装了ubuntu 16.04,但我无法获得caffe和tensorflow。笔记本电脑使用Nvidia Optimus技术在gpu和cpu之间切换以节省电池使用量。我检查了BIOS,看看我是否可以将它设置为仅使用gpu,但没有选项。使用大黄蜂或nvidia-prime也没有用。我现在有ubuntu 16与配合桌面环境,它阻止黑屏,但没有帮助解决cuda问题。我能够安装驱动程序和cuda,但是当我构建caffe和tensorflow时,他们没有说它没有检测到gpu。我无法安装opengl。我尝试使用几个版本的nvidia驱动程序,但它没有帮助。任何帮助都会很棒。感谢。

1 个答案:

答案 0 :(得分:0)

我认为Bumblebee可以让你在GPU模式下运行Caffe / Tensorflow。更一般地说,它还允许您使用Optimus技术在笔记本电脑上运行其他CUDA程序。

如果你正确安装了Bumblebee(教程:Bumblebee Wiki for Ubuntu),你可以通过在caffe二进制文件之前挂起optirun来调用Caffe二进制文件。所以它如下:

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调用的)用于选择驱动桌面的GPU。我有集成的GPU通过DisplayPort连接到HDMI端口和NVidia GPU的显示器。目前,DisplayPort处于非活动状态。显示信号来自HDMI端口。

据我所知,PRIME通过修改/etc/X11/Xorg.conf来使英特尔集成GPU或NVidia GPU成为X可用的当前显示适配器。我认为PRIME设置仅在两个GPU都有意义时才有意义连接到某个显示器,这意味着两个GPU之间不需要像笔记本电脑那样的Optimus链接(或者,对于带有Mux的笔记本电脑,如Dell Precision M4600,在BIOS中禁用Optimus)。

有关Display Mux和Optimus的更多信息,请访问:Using the NVIDIA Driver with Optimus Laptops

希望这有帮助!