ci: switch Docker image to nvidia/cuda:12.6.0-devel-rockylinux9 (#35)

* ci: switch to nvidia/cuda:12.6.0-devel-rockylinux9 cupy currently needs header files that aren't in the runtime image * ci: python3-devel
flatironinstitute · Aug 29, 2024 · 9cdd796 · 9cdd796
1 parent 4c85c30
commit 9cdd796
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ci/Dockerfile b/ci/Dockerfile
@@ -1,7 +1,7 @@
-FROM nvidia/cuda:12.3.2-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.0-devel-rockylinux9
 
-RUN apt-get update && \
-    DEBIAN_FRONTEND=noninteractive apt-get install -y \
+RUN dnf -y install \
     python3 \
+    python3-devel \
     python3-pip \
     git