Trying to compress steps to let github docker push successfully

uglabs · Apr 20, 2024 · b06facd · b06facd
1 parent 9c6eb95
commit b06facd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,3 +1,7 @@
 FROM archlinux
-RUN pacman -Syu --noconfirm
-RUN pacman -S --noconfirm python python-pip cudnn python-pytorch python-numpy
+RUN --mount=type=cache,sharing=locked,target=/var/cache/pacman pacman -Syu --noconfirm && pacman -S --noconfirm python python-pip cudnn python-pytorch python-numpy
+
+# configure nvidia container runtime
+# https://github.com/NVIDIA/nvidia-container-runtime#environment-variables-oci-spec
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility