setting which GPU to use

User 1554 | 3/16/2015, 7:17:58 PM

I have a machine with 2 GPUs and would like to choose which one to use when running a neuralnet classifier. How does one specify the device to use in GraphLab?

I'm looking for something akin to "export CUDAVISIBLEDEVICES=1" or cudaSetDevice()

Comments

User 1190 | 3/17/2015, 6:14:16 AM

Unfortunately, there is no way of selecting device, and it always uses device 0:

<pre><code> cudaSetDevice(0); </code></pre>

We will take your feature request and add multiple device support in the next release.