diff --git a/README.md b/README.md index 0f09b7840582c7aab0a79bb454ea0f29b2856469..a7e9f6137db329a9f7453a8dd6aa538b7b8f6845 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.