From c89964694f3ea1e33ea11caa12b6a811ff91269f Mon Sep 17 00:00:00 2001 From: Tomas Kracmar Date: Sun, 5 Oct 2025 12:03:58 +0200 Subject: [PATCH] Dockerfile fix --- app/Dockerfile | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/app/Dockerfile b/app/Dockerfile index 6f3de09..79fea15 100644 --- a/app/Dockerfile +++ b/app/Dockerfile @@ -1,7 +1,9 @@ ## GPU-ready base image with CUDA 12 + cuDNN 9 runtime # If you don't have an NVIDIA GPU or the NVIDIA Container Toolkit, this image still runs on CPU. # For smaller CPU-only images, you can switch back to python:3.11-slim. -FROM nvidia/cuda:12.4.1-cudnn9-runtime-ubuntu22.04 +# Use a widely available CUDA + cuDNN runtime tag. If you prefer newer CUDA, +# adjust this to a tag that exists on Docker Hub. +FROM nvidia/cuda:12.3.2-cudnn8-runtime-ubuntu22.04 # Keep python fast/quiet and pip lean ENV PYTHONDONTWRITEBYTECODE=1 \