search

coreweave / ml-containers

MIT License
19 stars 3 forks source link

build(torch): Build `nccl` images from bases with HPC-X v2.16 #36

Closed Eta0 closed 1 year ago

Eta0 commented 1 year ago

torch:nccl with HPC-X v2.16

This change updates the base nccl-tests images used to build torch:nccl to versions featuring HPC-X v2.16, as introduced in coreweave/nccl-tests#25.

github-actions[bot] commented 1 year ago

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/5979992174 Image: ghcr.io/coreweave/ml-containers/torch:es-hpc-x-2.16-3bdcc00-nccl-cuda11.8.0-nccl2.16.2-1-torch2.0.1-vision0.15.2-audio2.0.2