Upgrade to CUDA 12.1.0 to match pytorch version

sillsdev · Mar 20, 2024 · 82cabe2 · 82cabe2
1 parent ec6f04a
commit 82cabe2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-ARG CUDA_VERSION=11.2.2-cudnn8-runtime-ubuntu20.04
+ARG CUDA_VERSION=12.1.0-cudnn8-runtime-ubuntu20.04
 ARG PYTHON_VERSION=3.8
 ARG POETRY_VERSION=1.7.1
 

diff --git a/silnlp/nmt/clearml_connection.py b/silnlp/nmt/clearml_connection.py
@@ -49,7 +49,7 @@ def __post_init__(self) -> None:
 
             self.task.set_base_docker(
                 # docker_image="ghcr.io/sillsdev/silnlp:1.01.4",
-                docker_image="nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04",
+                docker_image="nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu20.04",
                 docker_arguments="-v /home/clearml/.clearml/hf-cache:/root/.cache/huggingface",
                 docker_setup_bash_script=[
                     "apt install -y python3-venv",