I currently have access to a system with two GPUs: GeForce GTX 1080 and Tesla K40c
I would like to use this machine for training neural networks in Mathematica.
When using the option
Target Device->"GPU" for
NetTrain it automatically chooses the GTX 1080 GPU for training (I can monitor this from the commandline using nvidia-smi).
However, this card doesn't work with Mathematica yet. It starts training but the loss quickly goes to zero. It's a common problem; see: http://community.wolfram.com/groups/-/m/t/917616
I would like to know how I can fix this issue or configure
NetTrain to use the other GPU (Tesla K40c).
$Version -> 11.0.1 for Linux x86 (64-bit) (September 21, 2016)