Dockerfile fix

This commit is contained in:
2025-10-05 12:03:58 +02:00
parent a450ab5c70
commit c89964694f

View File

@@ -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 \