upgrade deps to fix torch six bug (#329)

* update dependencies for mnist and pneumonia to pytorch 2.0 to avoid _six bug --------- Co-authored-by: Jeff Omhover <[email protected]> Co-authored-by: Thomas <[email protected]> Co-authored-by: thomasp-ms <[email protected]> Co-authored-by: unknown <[email protected]> Co-authored-by: Amit Garg <[email protected]>
Azure-Samples · Apr 5, 2023 · c024b67 · c024b67
1 parent d2bd683
commit c024b67
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/components/MNIST/traininsilo/conda.yaml b/examples/components/MNIST/traininsilo/conda.yaml
@@ -6,11 +6,11 @@ channels:
 dependencies:
   - python=3.8
   - pip=22.3.1
-  - pytorch=1.13.1
-  - torchvision=0.14.1
+  - pytorch=2.0.0
   - pytorch-cuda=11.6
   - pip:
     - azureml-mlflow==1.48.0
     - pandas==1.5.2
     - opacus==1.3.0
     - tqdm==4.64.1
+    - torchvision==0.15.1
diff --git a/examples/components/PNEUMONIA/traininsilo/conda.yaml b/examples/components/PNEUMONIA/traininsilo/conda.yaml
@@ -6,10 +6,10 @@ channels:
 dependencies:
   - python=3.8
   - pip=22.3.1
-  - pytorch=1.13.1
-  - torchvision=0.13.1
+  - pytorch=2.0.0
   - pytorch-cuda=11.6
   - pip:
     - azureml-mlflow==1.48.0
     - opacus==1.3.0
     - tqdm==4.64.1
+    - torchvision==0.15.1