From 4b57d533bfc03da7a7aae7a10ad0328d294fce8f Mon Sep 17 00:00:00 2001 From: "D. Kobus" Date: Fri, 29 Mar 2024 13:19:19 +0100 Subject: [PATCH 1/2] fix conda env --- conda-extras.yaml | 8 ++++---- conda.yaml | 12 ++++++------ 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/conda-extras.yaml b/conda-extras.yaml index 71574c4d3..db76d18d1 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 35dfc30ad..7e85de851 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 From 4547e69f19596f7c202be85a88015bd16bfa8234 Mon Sep 17 00:00:00 2001 From: "D. Kobus" Date: Fri, 29 Mar 2024 18:11:38 +0100 Subject: [PATCH 2/2] fix names and cuda label --- conda-extras.yaml | 2 +- conda.yaml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/conda-extras.yaml b/conda-extras.yaml index db76d18d1..c1ea3bfc7 100644 --- a/conda-extras.yaml +++ b/conda-extras.yaml @@ -2,9 +2,9 @@ name: dinov2-extras channels: - defaults - pytorch - - nvidia - xformers - conda-forge + - nvidia/label/cuda-11.7.0 dependencies: - python=3.10 - pytorch::pytorch=1.13.1 diff --git a/conda.yaml b/conda.yaml index 7e85de851..2ec394355 100644 --- a/conda.yaml +++ b/conda.yaml @@ -1,10 +1,10 @@ -name: dinov2-extras +name: dinov2 channels: - defaults - pytorch - - nvidia - xformers - conda-forge + - nvidia/label/cuda-11.7.0 dependencies: - python=3.10 - pytorch::pytorch=1.13.1