-
Notifications
You must be signed in to change notification settings - Fork 28
/
Dockerfile-gpu
33 lines (25 loc) · 773 Bytes
/
Dockerfile-gpu
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
FROM pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-runtime
WORKDIR /code
RUN apt-get update && \
apt-get install -y \
libglib2.0-0 \
libsm6 \
libxext6 \
libxrender-dev \
ffmpeg \
locales \
&& apt-get -y clean all \
&& rm -rf /var/lib/apt/lists/*
RUN locale-gen en_US.UTF-8
ENV LANG='en_US.UTF-8' LANGUAGE='en_US:en' LC_ALL='en_US.UTF-8'
COPY requirements-gpu.txt .
RUN pip install -r requirements-gpu.txt
RUN git clone https://github.com/alievk/first-order-model.git fomm
RUN cd fomm && git checkout efbe0a6f17b38360ff9a446fddfbb3ce5493534c && cd ..
WORKDIR /code
COPY download_model.py .
RUN python download_model.py
COPY app ./app
COPY afy ./afy
RUN mkdir app/static
CMD uvicorn app.server:app --port $PORT --workers 1 --host 0.0.0.0