From 3ea89dd605fcb98bb0d02d6cc516d8c9b24cb4c3 Mon Sep 17 00:00:00 2001 From: maurerv Date: Wed, 21 Feb 2024 00:23:30 +0100 Subject: [PATCH] Updated container version --- workflow/docker/pulldown.dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/workflow/docker/pulldown.dockerfile b/workflow/docker/pulldown.dockerfile index ba99187..8fbb433 100644 --- a/workflow/docker/pulldown.dockerfile +++ b/workflow/docker/pulldown.dockerfile @@ -14,7 +14,7 @@ # Extended by Valentin Maurer -ARG CUDA=11.1.1 +ARG CUDA=11.8 FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu18.04 ARG CUDA @@ -71,7 +71,7 @@ RUN pip3 install --upgrade pip --no-cache-dir \ pytest \ jax==0.4.23 \ jaxlib==0.4.23 \ - -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_cuda_releases.html + jaxlib==0.4.23+cuda11.cudnn86 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html RUN chmod u+s /sbin/ldconfig.real