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 \