forked from mlcommons/GaNDLF
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile-CPU
28 lines (25 loc) · 1.62 KB
/
Dockerfile-CPU
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
FROM ubuntu:18.04
LABEL github="https://github.com/CBICA/GaNDLF"
LABEL docs="https://cbica.github.io/GaNDLF/"
LABEL version=1.0
# Install fresh Python and dependencies for build-from-source
RUN apt-get update && apt-get install -y python3.8 python3-pip libvips libjpeg8-dev zlib1g-dev python3-dev libpython3.8-dev libffi-dev
RUN python3.8 -m pip install --upgrade pip
# EXPLICITLY install cpu versions of torch/torchvision (not all versions have +cpu modes on PyPI...)
RUN python3.8 -m pip install torch==1.8.2+cpu torchvision==0.9.2+cpu torchaudio===0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cpu
COPY . /GaNDLF
WORKDIR /GaNDLF
RUN python3.8 -m pip install openvino-dev==2022.1.0
RUN python3.8 -m pip install -e .
# Entrypoint forces all commands given via "docker run" to go through python, CMD forces the default entrypoint script argument to be gandlf_run
# If a user calls "docker run gandlf:[tag] gandlf_anonymize", it will resolve to running "python gandlf_anonymize" instead.
# CMD is inherently overridden by args to "docker run", entrypoint is constant.
ENTRYPOINT python3.8
CMD gandlf_run
# The below force the container commands to run as a nonroot user with UID > 10000.
# This greatly reduces UID collision probability between container and host, helping prevent privilege escalation attacks.
# As a side benefit this also decreases the likelihood that users on a cluster won't be able to access their files.
# See https://github.com/hexops/dockerfile as a best practices guide.
RUN addgroup --gid 10001 --system nonroot \
&& adduser --uid 10000 --system --ingroup nonroot --home /home/nonroot nonroot
USER nonroot