Skip to content

Python: caffe has no attribute 'NCCL' #318

@VasLem

Description

@VasLem

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions