https://github.com/Microsoft/CNTK
Revision 28f587f424396e1ad32f738eafbd2b968bd4b1c6 authored by Junjie Qian on 02 March 2018, 20:10:06 UTC, committed by Junjie Qian on 08 March 2018, 18:41:37 UTC
Thanks to Frank's initial code change.
Initialize NCCL with GPU ID frm nvml call instead of from device id in
/proc. Previous way causes "same device being used" error if two
different containers running on same host and try to initialize NCCL.

Add nvml library for nvml apis

Update the Makefile
1 parent dd88885
History
Tip revision: 28f587f424396e1ad32f738eafbd2b968bd4b1c6 authored by Junjie Qian on 02 March 2018, 20:10:06 UTC
Get GPU ID with nvmlDeviceGetUUID instead of device id in /proc/
Tip revision: 28f587f

back to top