dockerfile.gpu

FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04

WORKDIR /app
COPY . .
RUN apt-get update && apt-get install -y \
    python3 python3-dev python3-venv python3-pip \
    git gcc g++ curl && \
    python3 -m pip install --upgrade pip

RUN pip3 install torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
    pip3 install tensorflow==2.11.0 && \
    pip3 install -r requirements.txt

RUN find . -type f -name "*.pyc" -delete

EXPOSE 7860
CMD ["python3", "app.py"]

# รัน Docker กุขี้เกียจพิมพ์ เมื่อยนิ้วสัส
# docker build -f dockerfile.gpu -t energy_forecast .
# docker run --gpus=all -p 7861:7860 energy_forecast