From 82cabe2d5158fcc7962db465c4ff55e81c9d0662 Mon Sep 17 00:00:00 2001 From: Isaac Schifferer Date: Wed, 20 Mar 2024 17:34:59 -0400 Subject: [PATCH] Upgrade to CUDA 12.1.0 to match pytorch version --- Dockerfile | 2 +- silnlp/nmt/clearml_connection.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 14ef952e..7be01b20 100644 --- 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 index 17a99b2a..22f64b61 100644 --- 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",