Update Dockerfile

FeTS-AI · Feb 16, 2024 · d11351d · d11351d
1 parent 3df09a5
commit d11351d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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