Jenkins build

exoplanet-dev · Jun 23, 2024 · 628dcea · 628dcea
1 parent ef289e3
commit 628dcea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ci/Dockerfile b/ci/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.5.0-cudnn9-devel-ubuntu22.04
 
 RUN apt-get update && \
   DEBIAN_FRONTEND=noninteractive apt-get install -y \
@@ -14,4 +14,4 @@ RUN nvcc --version
 
 RUN python3 -m pip install -U pip && \
   python3 -m pip install -U pytest && \
-  python3 -m pip install -U "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+  python3 -m pip install -U "jax[cuda12]"
diff --git a/ci/Jenkinsfile b/ci/Jenkinsfile
@@ -17,7 +17,8 @@ pipeline {
         HOME = "$WORKSPACE/build"
       }
       steps {
-        sh 'EXOPLANET_CORE_CUDA=yes python3 -m pip install --user .'
+        sh 'EXOPLANET_CORE_CUDA=yes python3 -m pip install -v --user .'
+        sh 'python3 -c "import exoplanet_core.jax"'
         sh 'python3 -m pytest -v tests/jax_test.py'
       }
     }