diff --git a/conda-extras.yaml b/conda-extras.yaml index 71574c4d3..c1ea3bfc7 100644 --- a/conda-extras.yaml +++ b/conda-extras.yaml @@ -2,16 +2,16 @@ name: dinov2-extras channels: - defaults - pytorch - - nvidia - xformers - conda-forge + - nvidia/label/cuda-11.7.0 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 35dfc30ad..2ec394355 100644 --- a/conda.yaml +++ b/conda.yaml @@ -2,16 +2,16 @@ name: dinov2 channels: - defaults - pytorch - - nvidia - xformers - conda-forge + - nvidia/label/cuda-11.7.0 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