Try to resolve dgl on x86 and aarch64

Signed-off-by: Will Killian <wkillian@nvidia.com>
nv-morpheus · Feb 27, 2025 · 6639668 · 6639668
1 parent b44ecf9
commit 6639668
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 5 deletions.
diff --git a/conda/environments/all_cuda-128_arch-aarch64.yaml b/conda/environments/all_cuda-128_arch-aarch64.yaml
@@ -125,9 +125,10 @@ dependencies:
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.1/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl==2.2.1
   - faiss-cpu
   - google-search-results==2.4
   - langchain-nvidia-ai-endpoints==0.0.11

diff --git a/conda/environments/all_cuda-128_arch-x86_64.yaml b/conda/environments/all_cuda-128_arch-x86_64.yaml
@@ -131,9 +131,10 @@ dependencies:
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.1/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl==2.2.1
   - faiss-cpu
   - google-search-results==2.4
   - langchain-nvidia-ai-endpoints==0.0.11

diff --git a/conda/environments/examples_cuda-128_arch-aarch64.yaml b/conda/environments/examples_cuda-128_arch-aarch64.yaml
@@ -66,9 +66,10 @@ dependencies:
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.1/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl==2.2.1
   - faiss-cpu
   - google-search-results==2.4
   - langchain-nvidia-ai-endpoints==0.0.11

diff --git a/conda/environments/examples_cuda-128_arch-x86_64.yaml b/conda/environments/examples_cuda-128_arch-x86_64.yaml
@@ -68,9 +68,10 @@ dependencies:
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.1/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl==2.2.1
   - faiss-cpu
   - google-search-results==2.4
   - langchain-nvidia-ai-endpoints==0.0.11

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -537,7 +537,8 @@ dependencies:
           - pip
           - pip:
             - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
-            - dgl==2.4.0
+            - --find-links https://data.dgl.ai/wheels/torch-2.1/repo.html
+            - dgl==2.2.1
 
   example-abp-nvsmi:
     common: