pin pytorch-cuda to 11.8

Ensure we're installing the cuda-11.8 variant of dgl Update from pip style version spec to conda pytorch already in primary dev.yml
nv-morpheus · Aug 4, 2023 · 224d5d5 · 224d5d5
1 parent 6bb671a
commit 224d5d5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 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,22 +23,21 @@ 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
-    - torchmetrics==0.11.4
-    - tqdm==4.65.0
-    - torch==2.0.0+cu118
+    - torchmetrics=0.11.4
     - 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,10 @@ 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
+    - torchmetrics=0.11.4