Upgrade pytorch 1.13 to 2.2 (#3350)

Azure · Aug 15, 2024 · 07a31fc · 07a31fc
1 parent cc8649b
commit 07a31fc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/cli/jobs/single-step/pytorch/cifar-distributed/job.yml b/cli/jobs/single-step/pytorch/cifar-distributed/job.yml
@@ -11,7 +11,7 @@ inputs:
   cifar:
      type: uri_folder
      path: azureml:cifar-10-example@latest
-environment: azureml:AzureML-acpt-pytorch-1.13-cuda11.7@latest
+environment: azureml:AzureML-acpt-pytorch-2.2-cuda12.1@latest
 compute: azureml:gpu-cluster
 distribution:
   type: pytorch

diff --git a/cli/jobs/single-step/pytorch/word-language-model/job.yml b/cli/jobs/single-step/pytorch/word-language-model/job.yml
@@ -16,7 +16,7 @@ inputs:
   corpus:
     path: data
     mode: download
-environment: azureml:AzureML-acpt-pytorch-1.13-cuda11.7@latest
+environment: azureml:AzureML-acpt-pytorch-2.2-cuda12.1@latest
 compute: azureml:gpu-cluster
 display_name: pytorch-word-language-model-example
 experiment_name: pytorch-word-language-model-example

diff --git a/...perparameter-tune-deploy-with-pytorch/train-hyperparameter-tune-deploy-with-pytorch.ipynb b/...perparameter-tune-deploy-with-pytorch/train-hyperparameter-tune-deploy-with-pytorch.ipynb
@@ -199,7 +199,7 @@
    },
    "outputs": [],
    "source": [
-    "curated_env_name = \"AzureML-acpt-pytorch-1.13-cuda11.7@latest\""
+    "curated_env_name = \"AzureML-acpt-pytorch-2.2-cuda12.1@latest\""
    ]
   },
   {

diff --git a/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb b/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb
@@ -412,7 +412,7 @@
     "        # \"register_model_as\": \"places_dev\",\n",
     "        \"enable_profiling\": False,\n",
     "    },\n",
-    "    environment=\"AzureML-acpt-pytorch-1.13-cuda11.7@latest\",\n",
+    "    environment=\"AzureML-acpt-pytorch-2.2-cuda12.1@latest\",\n",
     "    compute=\"gpu-cluster\"\n",
     "    if (gpu_cluster)\n",
     "    else None,  # No compute needs to be passed to use serverless\n",