-
Notifications
You must be signed in to change notification settings - Fork 49
/
Dockerfile.rootfs
56 lines (46 loc) · 1.38 KB
/
Dockerfile.rootfs
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
FROM ubuntu:18.04
# Install some basic utilities
RUN apt-get update && apt-get install -y \
curl \
ca-certificates \
sudo \
git \
bzip2 \
libx11-6 \
&& rm -rf /var/lib/apt/lists/*
# Create a working directory
RUN mkdir /app
WORKDIR /app
# Create a non-root user and switch to it
RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
&& chown -R user:user /app
RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
USER user
# All users can use /home/user as their home directory
ENV HOME=/home/user
RUN chmod 777 /home/user
# Install Miniconda then use conda to install Python 3.8 and flask
ENV CONDA_AUTO_UPDATE_CONDA=false
ENV PATH=/home/user/miniconda/bin:$PATH
RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py38_4.8.2-Linux-x86_64.sh \
&& chmod +x ~/miniconda.sh \
&& ~/miniconda.sh -b -p ~/miniconda \
&& rm ~/miniconda.sh \
&& conda install -y python==3.8.1 \
&& conda install -y flask \
&& conda clean -ya
# Install pytorch, No CUDA-specific steps
ENV NO_CUDA=1
RUN conda install -y -c pytorch \
cpuonly \
"pytorch=1.7.1" \
"torchvision=0.8.2" \
"torchaudio=0.7.2" \
&& conda clean -ya
# Install onnxruntime
RUN conda install -c conda-forge onnxruntime
ADD test_samples /app
ADD src /app
RUN python3 download_pretrained_model.py \
&& rm /app/download_pretrained_model.py
CMD ["/bin/bash"]