gpu - theano - 仅将外部 GPU 用于 ML，将集成 GPU 用于显示

Question

我有一个带有集成 GPU 的 CPU。我还有一个用于 ML 的外部 GPU。我想要的是仅将集成 GPU 用于显示并将外部 GPU 专用于 NN 训练（以释放一些内存）。

我已在 BIOS 中将外部 GPU 设置为主 GPU，而且两者都处于活动状态。所以他们都在工作。启动系统后，我可以将显示器插入其中任何一个，它们都可以工作。

问题是当我将显示器插入主板（集成 GPU）时，theano 停止使用外部 GPU：

ERROR (theano.sandbox.cuda): ERROR: Not using GPU. Initialisation of device gpu failed:

有没有办法明确地将theano指向外部GPU？这是我的相关部分.theanorc：

[global]
floatX = float32
device = gpu

score 2 · Accepted Answer

我有一个和你类似的系统。对于 linux，安装bumblebee工作正常。

sudo apt-get install bumblebee-nvidia （适应你发行版的包管理器）

然后通过以下方式启动 python： optirun python

1 回答 1