diff --git a/Dockerfile b/Dockerfile index 0940e13..1c23e32 100644 --- a/Dockerfile +++ b/Dockerfile @@ -94,7 +94,8 @@ RUN /nnunet_env/bin/pip install git+https://github.com/MIC-DKFZ/nnUNet.git@nnune ENV nnUNet_raw_data_base="/tmp/nnUNet_raw_data_base" ENV nnUNet_preprocessed="/tmp/nnUNet_preprocessed" -ENV CUDA_VISIBLE_DEVICES="0" # see https://docs.docker.com/config/containers/resource_constraints/#gpu for detailed explanation +# see https://docs.docker.com/config/containers/resource_constraints/#gpu for detailed explanation +ENV CUDA_VISIBLE_DEVICES="0" COPY ./mlcubes/data_preparation/project /project