Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 18 commits into from
Jan 27, 2025
+82 −326
Merged

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

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
dedf8c2
Update the dockerfile base image so that we can support NCCL
Jan 14, 2025
ad2493a
remove no needed installs
Jan 14, 2025
d95d51c
fix installs and symlinks
Jan 15, 2025
d341fcd
forgot to take out tcpx
Jan 15, 2025
62000ec
fix way to link files, so cudnn.h is visible
Jan 15, 2025
62f6d65
Merge branch 'main' into sbosisio/cuda-dl-base
Steboss Jan 15, 2025
8b52c58
Address @olupton comments
Jan 15, 2025
6ff8cd4
Merge branch 'sbosisio/cuda-dl-base' of github.com:NVIDIA/JAX-Toolbox…
Jan 15, 2025
bd066f1
follow @olupton suggestion for nsys and check results
Jan 16, 2025
564ec47
Merge branch 'main' into sbosisio/cuda-dl-base
Steboss Jan 16, 2025
1154ccb
fix newline
Jan 17, 2025
bfb588d
add mods in container
Jan 21, 2025
65b5dd8
Address Olli's comments
Jan 21, 2025
5d2e464
remove DS_store
Jan 21, 2025
b058b15
remove check-shm
Jan 21, 2025
235ffac
update to the latest image
Jan 24, 2025
b2b5bcd
reset to 24.11 for nsight
Steboss Jan 27, 2025
ce1f3f3
Merge branch 'main' into sbosisio/cuda-dl-base
Steboss Jan 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/container/Dockerfile.base
View file
Open in desktop
Steboss marked this conversation as resolved.
Show resolved Hide resolved
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# syntax=docker/dockerfile:1-labs
ARG BASE_IMAGE=nvidia/cuda:12.6.3-devel-ubuntu24.04
ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:24.12-cuda12.6-devel-ubuntu24.04
ARG GIT_USER_NAME="JAX Toolbox"
ARG GIT_USER_EMAIL=jax@nvidia.com
ARG CLANG_VERSION=18
Expand Down
Loading