diff --git a/requirements-extras.txt b/requirements-extras.txt index ac75fb3eace10f58fa6b5bc373f24358394141e5..35aecabe23833b192f8523606e90916b1c7649f1 100644 --- a/requirements-extras.txt +++ b/requirements-extras.txt @@ -1,2 +1,5 @@ mmcv-full==1.5.0 mmsegmentation==0.27.0 +submitit +--extra-index-url https://pypi.nvidia.com +cuml-cu11 diff --git a/requirements.txt b/requirements.txt index 04c159c443b89330ff3c84257c41b011f9791257..69a421306ff5af00ca2e18a9b698e4690921748e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,6 +6,4 @@ torchmetrics==0.10.3 fvcore iopath xformers==0.0.18 -submitit ---extra-index-url https://pypi.nvidia.com -cuml-cu11 +