Back to torch 2.0.1

AutoResearch · Dec 11, 2023 · c5cd8d6 · c5cd8d6
1 parent e3c004a
commit c5cd8d6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/azureml/conda.yml b/azureml/conda.yml
@@ -15,4 +15,4 @@ dependencies:
     - xformers
     - scipy
     # This works, while installing from pytorch and cuda from conda does not
-    - torch==2.1.0    
+    - torch==2.0.1    
diff --git a/azureml/eval.yml b/azureml/eval.yml
@@ -33,6 +33,6 @@ environment: # azureml://registries/azureml/environments/acpt-pytorch-2.0-cuda11
   # image: nvcr.io/nvidia/pytorch:23.10-py3
   conda_file: conda.yml
 display_name: autodoc_prediction
-compute: azureml:v100cluster
+compute: azureml:t4cluster
 experiment_name: evaluation
 description: |
diff --git a/azureml/generate.yml b/azureml/generate.yml
@@ -23,6 +23,6 @@ environment:
   image: mcr.microsoft.com/azureml/curated/acpt-pytorch-2.0-cuda11.7:21
   conda_file: conda.yml
 display_name: autodoc_prediction
-compute: azureml:v100cluster
+compute: azureml:t4cluster
 experiment_name: prediction
 description: |
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,7 @@ dependencies = [
     "typer",
     "scipy",
     # This works, while installing from pytorch and cuda from conda does not",
-    "torch==2.1.0",
+    "torch==2.0.1",
     "transformers>=4.35.2",
 ]