forked from MIC-DKFZ/nnUNet
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Dockerfile
17 lines (14 loc) · 716 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# Contains pytorch, torchvision, cuda, cudnn
FROM nvcr.io/nvidia/pytorch:23.06-py3
MAINTAINER Pedro M. Gordaliza @petermcgor
#nnUnet is set up by default to look at the following dir. I mantain them for reproducibility
ARG resources="/opt/nnunet_resources"
ENV nnUNet_raw=$resources"/nnUNet_raw" nnUNet_preprocessed=$resources"/nnUNet_preprocessed" nnUNet_results=$resources"/nnUNet_results"
#Copy the files. TODO change to git clone https://github.com/PeterMcGor/nnUNet.git once the repo is tested the repo is tested
RUN mkdir /nnunet
WORKDIR /nnunet
COPY . .
## Install nnunet
RUN pip install -e .
## hiddenlayers for topologies
RUN pip install --upgrade git+https://github.com/FabianIsensee/hiddenlayer.git