FROMAGe / Dockerfile
alvanli
Added key back, will use gpu
806c000
raw
history blame
449 Bytes
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel as base
RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
ENV HOME=/exp/fromage
RUN apt-get update && apt-get -y install git
WORKDIR /exp/fromage
COPY ./requirements.txt ./requirements.txt
RUN python -m pip install -r ./requirements.txt
RUN python -m pip install gradio
COPY . .
RUN chmod -R a+rwX .
CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"]