From a4aed1d3025c7540c924fafc41997fe563978fc6 Mon Sep 17 00:00:00 2001
From: ochafik <ochafik@google.com>
Date: Sun, 19 Jan 2025 12:33:56 +0000
Subject: [PATCH] Update cuda.Dockerfile

---
 .devops/cuda.Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.devops/cuda.Dockerfile b/.devops/cuda.Dockerfile
index b1c14c17a..3c5cb0251 100644
--- a/.devops/cuda.Dockerfile
+++ b/.devops/cuda.Dockerfile
@@ -19,6 +19,8 @@ WORKDIR /app
 COPY . .
 
 RUN --mount=type=cache,target=/root/.ccache \
+    --mount=type=cache,target=/var/lib/apt/lists \
+    --mount=type=cache,target=/var/cache/apt \
     if [ "${CUDA_DOCKER_ARCH}" != "default" ]; then \
     export CMAKE_ARGS="-DCMAKE_CUDA_ARCHITECTURES=${CUDA_DOCKER_ARCH}"; \
     fi && \