Update versions in the build script

elixir-nx · Sep 14, 2023 · 2f7cd88 · 2f7cd88
1 parent bf437ae
commit 2f7cd88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/builds/build.sh b/builds/build.sh
@@ -33,8 +33,8 @@ case "$1" in
 
   "cuda120")
     docker build -t xla-cuda120 -f builds/cuda.Dockerfile \
-      --build-arg CUDA_VERSION=12.0.0 \
-      --build-arg CUDNN_VERSION=8.8.0 \
+      --build-arg CUDA_VERSION=12.1.0 \
+      --build-arg CUDNN_VERSION=8.9.0 \
       --build-arg XLA_TARGET=cuda120 \
       .