diff --git a/tutorials/accelerated-python/brev/dockerfile b/tutorials/accelerated-python/brev/dockerfile index 22a2a738..1db0338c 100644 --- a/tutorials/accelerated-python/brev/dockerfile +++ b/tutorials/accelerated-python/brev/dockerfile @@ -1,4 +1,4 @@ -FROM rapidsai/devcontainers:25.08-cpp-cuda12.8 +FROM ghcr.io/nvidia/mirrors/rapidsai-devcontainers-25.08-cpp-cuda12.8 ENV PIP_ROOT_USER_ACTION=ignore \ PIP_DISABLE_PIP_VERSION_CHECK=1 \ @@ -13,18 +13,16 @@ RUN apt-get update -y \ && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ git \ git-lfs \ - && apt-get clean -y \ - && rm -rf /var/lib/apt/lists/* - -# Install Docker -RUN apt-get update -y \ - && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ apt-transport-https \ ca-certificates \ curl \ gnupg \ lsb-release \ - && mkdir -p /etc/apt/keyrings \ + && apt-get clean -y \ + && rm -rf /var/lib/apt/lists/* + +# Install Docker +RUN mkdir -p /etc/apt/keyrings \ && curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg \ && echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null \ && apt-get update -y \ diff --git a/tutorials/accelerated-python/brev/requirements.txt b/tutorials/accelerated-python/brev/requirements.txt index b032de71..894481dc 100644 --- a/tutorials/accelerated-python/brev/requirements.txt +++ b/tutorials/accelerated-python/brev/requirements.txt @@ -13,15 +13,15 @@ jupyterlab-nvidia-nsight jupyterlab-execute-time # CUDA -cuda-python -numba-cuda == 0.20.* +cuda-python == 12.9.* +numba-cuda nvmath-python[cu12] cupy-cuda12x -cuda-cccl[test-cu12] == 0.3.1 -nvidia-cuda-runtime-cu12 == 12.8.* +cuda-cccl[test-cu12] +nvidia-cublas-cu12 == 12.8.4.1 +nvidia-nvshmem-cu12 == 3.3.20 nvidia-cuda-nvcc-cu12 == 12.8.* nvidia-cuda-nvrtc-cu12 == 12.8.* -nvidia-nvjitlink-cu12 == 12.8.* # NVIDIA devtools nvtx