From cba717eb3be1dc4b491628bdb512bd9b3831cc06 Mon Sep 17 00:00:00 2001 From: Xuan Gu <xuan.gu@liu.se> Date: Tue, 30 May 2023 18:52:28 +0000 Subject: [PATCH] Update file README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 0f09b78..a7e9f61 100644 --- a/README.md +++ b/README.md @@ -13,6 +13,7 @@ cd ~/DeepLearningExamples/nnunet_for_pytorch_$VERSION/ ``` - Build the nnU-Net PyTorch Docker container + Change the pytorch-lightning version to 1.5.10 to avoid the `from torchmetrics.utilities.data import get_num_classes as _get_num_classes` error. Add `ENV PYTHONNOUSERSITE=True` to the `Dockerfile` to disable the user packages. -- GitLab