@@ -49,6 +49,9 @@ When running a training inside the container it is necessary to [increase the sh
...
@@ -49,6 +49,9 @@ When running a training inside the container it is necessary to [increase the sh
## Source
## Source
*Please note that nndetection requires Python 3.8+.*
1. Install CUDA (>10.1) and cudnn (make sure to select [compatible versions](https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html)!)
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.
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.
3. Install [torch](https://pytorch.org/)(make sure to match the pytorch and CUDA versions!) (requires pytorch >1.7+) and [torchvision](https://github.com/pytorch/vision)(make sure to match the versions!).
3. Install [torch](https://pytorch.org/)(make sure to match the pytorch and CUDA versions!) (requires pytorch >1.7+) and [torchvision](https://github.com/pytorch/vision)(make sure to match the versions!).
...
@@ -185,7 +188,7 @@ Note: Please avoid `.` inside file names since it can influence how paths/names
...
@@ -185,7 +188,7 @@ Note: Please avoid `.` inside file names since it can influence how paths/names