Spaces:
Running
Running
File size: 716 Bytes
b539248 09283ec b539248 09283ec b539248 c06b0aa b539248 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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 |