diff --git a/PyTorch/Segmentation/nnUNet/benchmark.txt b/PyTorch/Segmentation/nnUNet/benchmark.txt
index 9257eb6656cf478af225908f702085d51c93f2f5..55b265bb253878925c8c08036655658c7217d3ca 100644
--- a/PyTorch/Segmentation/nnUNet/benchmark.txt
+++ b/PyTorch/Segmentation/nnUNet/benchmark.txt
@@ -33,4 +33,6 @@ Traceback (most recent call last):
     trainer.current_epoch = 1
 AttributeError: can't set attribute
 
+Solution: pip install torchmetrics==0.6.0, works.
+
 # 2023-09-28: Not working on the devel reservation
\ No newline at end of file