update

pytorch · Mar 8, 2024 · 7d7349a · 7d7349a
1 parent a962486
commit 7d7349a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.ci/docker/build.sh b/.ci/docker/build.sh
@@ -20,5 +20,8 @@ docker build \
   --no-cache \
   --progress=plain \
   -f Dockerfile \
+  --build-arg BASE_IMAGE="${BASE_IMAGE}" \
+  --build-arg CUDA_VERSION="${CUDA_VERSION}" \
+  --build-arg UBUNTU_VERSION="${UBUNTU_VERSION}" \
   "$@" \
   .