【发布时间】:2017-04-04 18:36:01
【问题描述】:
我有一个带有集成 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
【问题讨论】:
标签: gpu theano theano-cuda