From 7fadf1389f3f5e711563aa9865b84feb516ff627 Mon Sep 17 00:00:00 2001 From: jason-e-gross Date: Sat, 20 Sep 2025 17:48:51 -0500 Subject: [PATCH 1/3] Update CUDA base image to version 12.8.1 --- Dockerfile | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 820ae94a..ed0be490 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,7 @@ -FROM nvidia/cuda:12.4.1-base-ubuntu22.04 +#Original +#FROM nvidia/cuda:12.4.1-base-ubuntu22.04 +#Cuda 12.8/13 +FROM nvidia/cuda:12.8.1-base-ubuntu22.04 ENV DEBIAN_FRONTEND noninteractive ENV CMDARGS --listen From 07fb4c96a47f85c77e1167107395b34599220da7 Mon Sep 17 00:00:00 2001 From: jason-e-gross Date: Sat, 20 Sep 2025 17:49:22 -0500 Subject: [PATCH 2/3] Update torch and torchvision versions --- requirements_docker.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements_docker.txt b/requirements_docker.txt index 21883adf..dde4431a 100644 --- a/requirements_docker.txt +++ b/requirements_docker.txt @@ -1,2 +1,2 @@ -torch==2.1.0 -torchvision==0.16.0 +torch==2.8.0 +torchvision==0.23.0 From 9d97e30b99b773502d4755fe4347974d5add9dae Mon Sep 17 00:00:00 2001 From: Manuel Schmid <9307310+mashb1t@users.noreply.github.com> Date: Thu, 25 Sep 2025 20:46:31 +0200 Subject: [PATCH 3/3] refactor: remove comment from Dockerfile --- Dockerfile | 3 --- 1 file changed, 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index ed0be490..bc3a3612 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,6 +1,3 @@ -#Original -#FROM nvidia/cuda:12.4.1-base-ubuntu22.04 -#Cuda 12.8/13 FROM nvidia/cuda:12.8.1-base-ubuntu22.04 ENV DEBIAN_FRONTEND noninteractive ENV CMDARGS --listen