@@ -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.
