Try to resolve dgl on x86 and aarch64

Signed-off-by: Will Killian <wkillian@nvidia.com>
nv-morpheus · Feb 27, 2025 · 9cce75d · 9cce75d
1 parent b44ecf9
commit 9cce75d
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
@@ -124,10 +124,11 @@ dependencies:
 - yapf=0.43
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
+  - --find-links https://data.dgl.ai/wheels/torch-2.3/repo.html
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl
   - 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
@@ -130,10 +130,11 @@ dependencies:
 - yapf=0.43
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
+  - --find-links https://data.dgl.ai/wheels/torch-2.3/repo.html
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl
   - 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
@@ -65,10 +65,11 @@ dependencies:
 - websockets
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
+  - --find-links https://data.dgl.ai/wheels/torch-2.3/repo.html
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl
   - 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
@@ -67,10 +67,11 @@ dependencies:
 - websockets
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
+  - --find-links https://data.dgl.ai/wheels/torch-2.3/repo.html
   - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.4.0
+  - dgl
   - faiss-cpu
   - google-search-results==2.4
   - langchain-nvidia-ai-endpoints==0.0.11

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -536,8 +536,9 @@ dependencies:
           - &cuml cuml=25.02.*
           - pip
           - pip:
+            - --find-links https://data.dgl.ai/wheels/torch-2.3/repo.html
             - --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
-            - dgl==2.4.0
+            - dgl
 
   example-abp-nvsmi:
     common: