theano - 仅将外部GPU用于ML和集成GPU进行显示

时间:2016-11-21 11:21:54

标签: gpu theano theano-cuda

我有一个集成了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

1 个答案:

答案 0 :(得分:2)

我和你的系统类似。对于linux,安装bumblebee有效。

sudo apt-get install bumblebee-nvidia (适应你的发行包经理)

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