forked from BVLC/caffe
-
Notifications
You must be signed in to change notification settings - Fork 260
Closed
Description
I guess this is a matter of different versions, but since January (BVLC#4563) caffe from BVLC has a NCCL module inside _caffe.py . I dived inside Python wrappers but I did not find any such wrapper for NCCL inside nv-caffe. Are there any plans to implement this module ? I ask, because CUDA users face a problem using NCCL module from original caffe, whereas NCCL in nv-caffe is tailored for CUDA (as expected for a NVIDIA library). Furthermore, by using such module, one could do multiGPU training using Python code.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels