Merge pull request #1 from dagardner-nv/david-fsi_dgl-patch

Update dependencies
nv-morpheus · Aug 4, 2023 · d26d4b5 · d26d4b5
2 parents 9c6c7b3 + ee8e796
commit d26d4b5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/docker/conda/environments/cuda11.8_dev.yml b/docker/conda/environments/cuda11.8_dev.yml
@@ -83,7 +83,7 @@ dependencies:
     - python-graphviz
     - python=3.10
     - pytorch=2.0.1
-    - pytorch-cuda
+    - pytorch-cuda=11.8
     - rapidjson=1.1.0
     - scikit-build=0.17.1
     - scikit-learn=1.2.2

diff --git a/docker/conda/environments/cuda11.8_examples.yml b/docker/conda/environments/cuda11.8_examples.yml
@@ -23,20 +23,20 @@ channels:
     - rapidsai
     - nvidia
     - conda-forge
+    - dglteam/label/cu118
 dependencies:
     - boto3
     - chardet=5.0.0
     - cuml=23.06
     - dask>=2023.1.1
+    - dgl=1.0.2
     - dill=0.3.6
     - distributed>=2023.1.1
     - mlflow>=2.2.1,<3
     - papermill=2.3.4
     - s3fs>=2023.6
     - pip
     - wrapt=1.14.1 # ver 1.15 breaks the keras model used by the gnn_fraud_detection_pipeline
-    - dgl==1.0.2+cu118
-    - torch==2.0.0+cu118
     - pip:
         # tensorflow exists in conda-forge but is tied to CUDA-11.3
         - tensorflow==2.12.0
diff --git a/examples/gnn_fraud_detection_pipeline/requirements.yml b/examples/gnn_fraud_detection_pipeline/requirements.yml
@@ -17,11 +17,9 @@ channels:
     - rapidsai
     - nvidia
     - conda-forge
+    - dglteam/label/cu118
 dependencies:
     - cuml=23.06
     - dask>=2023.1.1
+    - dgl=1.0.2
     - distributed>=2023.1.1
-    - dgl==1.0.2+cu118
-    - torchmetrics==0.11.4
-    - tqdm==4.65.0
-    - torch==2.0.0+cu118