Update the dockerfile base image to cuda-dl-base#1248

Merged

Steboss merged 18 commits intomainfrom

sbosisio/cuda-dl-base

Jan 27, 2025

Commits on Jan 14, 2025

Update the dockerfile base image so that we can support NCCL
STEFANO BOSISIO
committed
remove no needed installs
STEFANO BOSISIO
committed

Commits on Jan 15, 2025

fix installs and symlinks
STEFANO BOSISIO
committed
forgot to take out tcpx
STEFANO BOSISIO
committed
fix way to link files, so cudnn.h is visible
STEFANO BOSISIO
committed
Merge branch 'main' into sbosisio/cuda-dl-base
Steboss
authored
Address @olupton comments
STEFANO BOSISIO
committed
Merge branch 'sbosisio/cuda-dl-base' of github.com:NVIDIA/JAX-Toolbox into sbosisio/cuda-dl-base
STEFANO BOSISIO
committed

Commits on Jan 16, 2025

follow @olupton suggestion for nsys and check results
STEFANO BOSISIO
committed
Merge branch 'main' into sbosisio/cuda-dl-base
Steboss
authored

Commits on Jan 17, 2025

fix newline
STEFANO BOSISIO
committed

Commits on Jan 21, 2025

add mods in container
STEFANO BOSISIO
committed
Address Olli's comments
STEFANO BOSISIO
committed
remove DS_store
STEFANO BOSISIO
committed
remove check-shm
STEFANO BOSISIO
committed

Commits on Jan 24, 2025

update to the latest image
STEFANO BOSISIO
committed

Commits on Jan 27, 2025