| # Base image with CUDA 12.8, cuDNN 9, PyTorch 2.7.0 | |
| FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-runtime | |
| # Install system tools (optional, like git, wget) | |
| RUN apt-get update && apt-get install -y \ | |
| git \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Set working directory | |
| WORKDIR /app | |
| # Copy and install Python dependencies | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| # Default command: open Python shell | |
| CMD ["python"] | |