Closed bchandle closed 8 years ago
I have eliminated the auto-pruning of "weak GPUs" that have less local memory than the best GPU on the platform (available in release v5.0.0-alpha.4). As before, choose the device you desire with the -Dcog.device=N JVM argument, or in your scala code as in:
val cg = new ComputeGraph(device=N) { ... }
Please verify this fix works for you.
Yes, the fix works. The test suite isn't passing on either the Iris Pro or GT 750M right now, but I'll open separate issues for those problems.
I'm seeing this behavior on a MacBook Pro with a discreet NVIDIA GPU. This is the debugging output with verboseOpenCLDevices and verboseOpenCLPlatform enabled:
The problem appears to be the GPU device selection code in
OpenCLPlatform.createDevices()
:The Iris Pro is a weaker GPU, but it's returning 65536 for
getLocalMemSize
compared to 49152 from the GT 750M. The CCT device selection logic is ignoring the NVIDIA GPU because of the comparatively smaller local memory.@DickJC123, any suggestions on how to proceed? The simplest fix for this particular platform would be to return all GPU devices and let the user pick.