Update to mppmu/julia-conda:ub22-jl111-mf-cu126

Also update ML packages

Author: oschulz

Dockerfile

@@ -1,4 +1,4 @@
-FROM mppmu/julia-conda:ub22-jl111-mf-cu124
+FROM mppmu/julia-conda:ub22-jl111-mf-cu126
 
 # User and workdir settings:
 
@@ -136,16 +136,16 @@ RUN true \
 # Install PyTorch:
 
 # Need to use pip to make PyTorch uses system-wide CUDA libs:
-RUN pip3 install --upgrade \
-    torch~=2.5.1 \
+RUN pip3 install --upgrade --index-url https://download.pytorch.org/whl/cu126 \
+    torch~=2.6.0 \
     torchvision \
     torchaudio
 
 
 # Install JAX:
 
 RUN pip3 install --upgrade \
-    "jax[cuda12]~=0.5.0"
+    "jax[cuda12]~=0.5.2"
 
 
 # Install dcraw and ImageMagick

README.md

@@ -6,10 +6,10 @@ The Docker image includes the following Software:
 
 * Python via miniforge (Python 3.12)
 * CERN ROOT v6.28
-* CUDA v12.1
+* CUDA v12.6
 * Geant4 v10.5 and CLHep
 * HDF5 (thread-safe build)
-* Julia v1.6, v1.10 and v1.11 (default)
+* Julia v1.10 and v1.11 (default)
 * PyTorch and JAX
 * Snakemake with panoptes web-GUI
 * Node.js