Skip to content

Commit

Permalink
add Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
PeterMcGor committed Dec 3, 2023
1 parent 2bc504d commit 34ce4b2
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,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

Check failure on line 5 in Dockerfile

View workflow job for this annotation

GitHub Actions / Check for spelling errors

mantain ==> maintain
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

0 comments on commit 34ce4b2

Please sign in to comment.