diff --git a/conda-extras.yaml b/conda-extras.yaml index 71574c4..db76d18 100644 --- a/conda-extras.yaml +++ b/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 diff --git a/conda.yaml b/conda.yaml index 35dfc30..7e85de8 100644 --- a/conda.yaml +++ b/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 \ No newline at end of file