search

coreweave / ml-containers

MIT License
19 stars 3 forks source link

build(torch): Update base images for NCCL builds, now with NCCL v2.18.3 #32

Closed Eta0 closed 1 year ago

Eta0 commented 1 year ago

PyTorch with NCCL v2.18.3

This change uses the new NCCL v2.18.3 builds from coreweave/nccl-tests PR 23 in this repository's CUDA 12.0 & 12.1 builds of torch:nccl, updating from NCCL version 2.18.1.

github-actions[bot] commented 1 year ago

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/5662813899 Image: ghcr.io/coreweave/ml-containers/torch:es-torch-nccl-2.18.3-f6ea142-nccl-cuda12.0.1-nccl2.18.3-1-torch2.0.1-vision0.15.2-audio2.0.2

github-actions[bot] commented 1 year ago

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/5662813899 Image: ghcr.io/coreweave/ml-containers/torch:es-torch-nccl-2.18.3-f6ea142-nccl-cuda11.8.0-nccl2.16.2-1-torch2.0.1-vision0.15.2-audio2.0.2

github-actions[bot] commented 1 year ago

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/5662813899 Image: ghcr.io/coreweave/ml-containers/torch:es-torch-nccl-2.18.3-f6ea142-nccl-cuda12.1.1-nccl2.18.3-1-torch2.0.1-vision0.15.2-audio2.0.2

github-actions[bot] commented 1 year ago

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/5673037243 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-torch-nccl-2.18.3-e7210fd-nccl-cuda12.0.1-nccl2.18.3-1-torch2.0.1-vision0.15.2-audio2.0.2