Skip to main content
NVIDIA
CUDA Deep Learning
Container
NVIDIA
CUDA Deep Learning

CUDA is a parallel computing platform and programming model that enhances computing performance using NVIDIA GPUs. CUDA Deep Learning integrates networking and GPU-accelerated libraries like cuDNN, cuTensor, NCCL, HPC-x, and the CUDA Toolkit.

  • LayerLabelCreated
    sha256:bb8bdfe4c2f8ccbb8ee55e33723968ba9952eac098fd2715c40b5cf3ad08f927RUN
    RUN |5 ENABLE_MITMPROXY=0 VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION LIBFABRIC_INSTALL_PATH=/opt/amazon/efa RDMA_CORE_PREFIX=/usr TARGETARCH=amd64 /bin/sh -c set -exo pipefail if [ "${ENABLE_MITMPROXY}" = "1" ]; then /tmp/manage_cert.sh install; fi mkdir -p /usr/local/nixl cp -a /nixl/usr/local/nixl/. /usr/local/nixl/ ls -lart /usr/local/nixl # Copy Abseil-CPP library cp -a /nixl/usr/local/lib/libabsl* /usr/local/lib/ # Copy gRPC library cp -a /nixl/usr/local/lib/libutf8* /usr/local/lib/ cp -a /nixl/usr/local/lib/libprotobuf* /usr/local/lib/ cp -a /nixl/usr/local/lib/libprotoc* /usr/local/lib/ cp -a /nixl/usr/local/include/utf8* /usr/local/include/ mkdir -p /usr/local/include/{google,upb,re2,grpc,grpc++,grpcpp} cp -a /nixl/usr/local/include/google/. /usr/local/include/google/ cp -a /nixl/usr/local/include/upb/. /usr/local/include/upb/ cp -a /nixl/usr/local/include/re2/. /usr/local/include/re2/ cp -a /nixl/usr/local/lib/libre2* /usr/local/lib/ cp -a /nixl/usr/local/lib/libaddress* /usr/local/lib/ cp -a /nixl/usr/local/include/grpc/. /usr/local/include/grpc/ cp -a /nixl/usr/local/lib/libgpr* /usr/local/lib/ cp -a /nixl/usr/local/lib/libgrpc* /usr/local/lib/ cp -a /nixl/usr/local/lib/libupb* /usr/local/lib/ cp -a /nixl/usr/local/include/grpc++/. /usr/local/include/grpc++/ cp -a /nixl/usr/local/include/grpcpp/. /usr/local/include/grpcpp/ # Copy etcd library cp /nixl/usr/local/lib/libetcd-cpp-api-core.so /usr/local/lib/libetcd-cpp-api-core.so # Copy aws-sdk-cpp headers mkdir -p /usr/local/include/aws cp -a /nixl/usr/local/include/aws/. /usr/local/include/aws/ cp -a /nixl/usr/local/lib/libaws-* /usr/local/lib/ ls -lart /usr/local/include/aws # Copy gusli headers cp -a /nixl/usr/include/gus* /usr/include/ cp -a /nixl/usr/lib/libg* /usr/lib/ # Copy nixlbench mkdir -p /usr/local/nixlbench/bin /usr/local/nixlbench/include /usr/local/nixlbench/lib cp -a /nixl/usr/local/nixlbench/. /usr/local/nixlbench/ ls -lart /usr/local/nixlbench # Install nixlbench runtime dependencies apt update apt install -y libgflags-dev libaio1t64 liburing2 rm -rf /var/lib/apt/lists/* # Copy s2n-tls headers and libs cp -a /nixl/usr/local/s2n-tls-install/include/. /usr/local/include/ cp -a /nixl/usr/local/s2n-tls-install/lib/. /usr/local/lib/ ls -lart /usr/local/lib/libs2n* ldconfig if [ "${ENABLE_MITMPROXY}" = "1" ]; then /tmp/manage_cert.sh uninstall; fi # buildkit
    05/30/2026 6:34 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TARGETARCH=amd64
    05/30/2026 6:25 PM UTC
    sha256:099b697deda0b8194ad4c358cbd0157a892237ce769ad1044d8ecf5676406051COPY
    ./common.sh ./common.sh
    05/30/2026 6:25 PM UTC
    sha256:75bb374681dfe58bee80b5e393d84fe5570d4d0e4b9c127d32b82989a459fc28COPY
    ./test_nixlbench.sh ./test_nixlbench.sh
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/nixlbench/lib:/usr/local/lib:/opt/amazon/efa/lib:/usr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nixl/lib/x86_64-linux-gnu:/usr/local/nixl/lib/aarch64-linux-gnu
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/usr/local/nixlbench/bin:/usr/local/nixl/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    ENABLE_MITMPROXY=0
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/lib:/opt/amazon/efa/lib:/usr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nixl/lib/x86_64-linux-gnu:/usr/local/nixl/lib/aarch64-linux-gnu
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    RDMA_CORE_PREFIX=/usr
    05/30/2026 6:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    LIBFABRIC_INSTALL_PATH=/opt/amazon/efa
    05/30/2026 6:25 PM UTC
    ...