Bump dgl version and use newer wheels repo URL

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

Hide file tree

Showing 5 changed files with 10 additions and 15 deletions.
diff --git a/conda/environments/all_cuda-128_arch-aarch64.yaml b/conda/environments/all_cuda-128_arch-aarch64.yaml
@@ -124,11 +124,10 @@ dependencies:
 - yapf=0.43
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
-  - --find-links https://data.dgl.ai/wheels-test/repo.html
-  - --find-links https://data.dgl.ai/wheels/cu121/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.4/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.1.0
+  - dgl==2.4.0
   - dglgo
   - faiss-cpu
   - google-search-results==2.4

diff --git a/conda/environments/all_cuda-128_arch-x86_64.yaml b/conda/environments/all_cuda-128_arch-x86_64.yaml
@@ -130,11 +130,10 @@ dependencies:
 - yapf=0.43
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
-  - --find-links https://data.dgl.ai/wheels-test/repo.html
-  - --find-links https://data.dgl.ai/wheels/cu121/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.4/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.1.0
+  - dgl==2.4.0
   - dglgo
   - faiss-cpu
   - google-search-results==2.4

diff --git a/conda/environments/examples_cuda-128_arch-aarch64.yaml b/conda/environments/examples_cuda-128_arch-aarch64.yaml
@@ -65,11 +65,10 @@ dependencies:
 - websockets
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
-  - --find-links https://data.dgl.ai/wheels-test/repo.html
-  - --find-links https://data.dgl.ai/wheels/cu121/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.4/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.1.0
+  - dgl==2.4.0
   - dglgo
   - faiss-cpu
   - google-search-results==2.4

diff --git a/conda/environments/examples_cuda-128_arch-x86_64.yaml b/conda/environments/examples_cuda-128_arch-x86_64.yaml
@@ -67,11 +67,10 @@ dependencies:
 - websockets
 - pip:
   - --extra-index-url https://download.pytorch.org/whl/cu124
-  - --find-links https://data.dgl.ai/wheels-test/repo.html
-  - --find-links https://data.dgl.ai/wheels/cu121/repo.html
+  - --find-links https://data.dgl.ai/wheels/torch-2.4/repo.html
   - databricks-cli < 0.100
   - databricks-connect
-  - dgl==2.1.0
+  - dgl==2.4.0
   - dglgo
   - faiss-cpu
   - google-search-results==2.4

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