/nccl

golang wrapper for nccl

Primary LanguageGoMIT LicenseMIT

nccl

golang wrapper for nccl

nccl version 2.4.8

nccl needs to be downloaded from nvidia.

You might need to find where nccl is installed to use this. It took me a little while to find where the libnccl and nccl.h was located. I had to find it in some APT log. I can't quite remember where the location of that was.