fix conda env

conda-extras.yaml

@@ -6,12 +6,12 @@ channels:
   - xformers
   - conda-forge
 dependencies:
-  - python=3.9
-  - pytorch::pytorch=2.0.0
+  - python=3.10
+  - pytorch::pytorch=1.13.1
   - pytorch::pytorch-cuda=11.7.0
-  - pytorch::torchvision=0.15.0
+  - pytorch::torchvision
   - omegaconf
-  - torchmetrics=0.10.3
+  - torchmetrics
   - fvcore
   - iopath
   - xformers::xformers=0.0.18

conda.yaml

@@ -1,4 +1,4 @@
-name: dinov2
+name: dinov2-extras
 channels:
   - defaults
   - pytorch
@@ -6,12 +6,12 @@ channels:
   - xformers
   - conda-forge
 dependencies:
-  - python=3.9
-  - pytorch::pytorch=2.0.0
+  - python=3.10
+  - pytorch::pytorch=1.13.1
   - pytorch::pytorch-cuda=11.7.0
-  - pytorch::torchvision=0.15.0
+  - pytorch::torchvision
   - omegaconf
-  - torchmetrics=0.10.3
+  - torchmetrics
   - fvcore
   - iopath
   - xformers::xformers=0.0.18
@@ -19,4 +19,4 @@ dependencies:
   - pip:
     - git+https://github.com/facebookincubator/submitit
     - --extra-index-url https://pypi.nvidia.com
-    - cuml-cu11
+    - cuml-cu11