forked from MIC-DKFZ/nnDetection
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
49 lines (42 loc) · 1.51 KB
/
Dockerfile
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
#Copyright 2020 Division of Medical Image Computing, German Cancer Research Center (DKFZ), Heidelberg, Germany
#
#Licensed under the Apache License, Version 2.0 (the "License");
#you may not use this file except in compliance with the License.
#You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
#Unless required by applicable law or agreed to in writing, software
#distributed under the License is distributed on an "AS IS" BASIS,
#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#See the License for the specific language governing permissions and
#limitations under the License.
# Contains pytorch, torchvision, cuda, cudnn
FROM nvcr.io/nvidia/pytorch:20.12-py3
ARG env_det_num_threads=6
ARG env_det_verbose=1
# Setup environment variables
ENV det_data=/opt/data det_models=/opt/models det_num_threads=$env_det_num_threads det_verbose=$env_det_verbose OMP_NUM_THREADS=1
# Install some tools
RUN apt-get update && export DEBIAN_FRONTEND=noninteractive && apt-get install -y \
git \
cmake \
make \
wget \
gnupg \
build-essential \
software-properties-common \
gdb \
ninja-build
RUN pip install numpy
# Install own code
COPY ./requirements.txt .
RUN mkdir ${det_data} \
&& mkdir ${det_models} \
&& mkdir -p /opt/code/nndet \
&& pip install -r requirements.txt \
&& pip install hydra-core --upgrade --pre \
&& pip install git+https://github.com/mibaumgartner/pytorch_model_summary.git
WORKDIR /opt/code/nndet
COPY . .
RUN FORCE_CUDA=1 pip install -v -e .