Update README.md

README.md

@@ -51,6 +51,7 @@ When running a training inside the container it is necessary to [increase the sh
 ## Source
 
 *Please note that nndetection requires Python 3.8+.*
+*Please use PyTorch 1.X version for now and not 2.0*
 
 1. Install CUDA (>10.1) and cudnn (make sure to select [compatible versions](https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html)!)
 2. [Optional] Depending on your GPU you might need to set `TORCH_CUDA_ARCH_LIST`, check [compute capabilities](https://developer.nvidia.com/cuda-gpus) here.