我有一个集成了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
答案 0 :(得分:2)
我和你的系统类似。对于linux,安装bumblebee
有效。
sudo apt-get install bumblebee-nvidia
(适应你的发行包经理)
然后通过以下方式启动python:
optirun python