Updated tensorflow-2.16-cuda11 to tensorflow-2.16-cuda12 (#3436)

Azure · Oct 31, 2024 · 28af259 · 28af259
1 parent 0f429c8
commit 28af259
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/cli/jobs/basics/hello-interactive.yml b/cli/jobs/basics/hello-interactive.yml
@@ -1,7 +1,7 @@
 $schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
 command: python hello-interactive.py && sleep 600
 code: src
-environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda12@latest
 
 services:
   my_vscode:

diff --git a/cli/jobs/pipelines-with-components/basics/6a_tf_hello_world/component.yml b/cli/jobs/pipelines-with-components/basics/6a_tf_hello_world/component.yml
@@ -7,7 +7,7 @@ name: tf_hello_world
 display_name: TF-hello-world
 version: 1
 
-environment: azureml://registries/azureml/environments/tensorflow-2.16-cuda11/labels/latest
+environment: azureml://registries/azureml/environments/tensorflow-2.16-cuda12/labels/latest
 command: echo $TF_CONFIG
 resources:
   instance_count: 2

diff --git a/cli/jobs/pipelines/tensorflow-image-segmentation/pipeline.yml b/cli/jobs/pipelines/tensorflow-image-segmentation/pipeline.yml
@@ -44,7 +44,7 @@ jobs:
       type: tensorflow
       worker_count: 1 # needs to match instance_count (!)
 
-    environment: azureml://registries/azureml/environments/tensorflow-2.16-cuda11/labels/latest
+    environment: azureml://registries/azureml/environments/tensorflow-2.16-cuda12/labels/latest
     # uncomment below to use custom environment
     # environment:
     #   build: 

diff --git a/cli/jobs/single-step/tensorflow/mnist-distributed/job.yml b/cli/jobs/single-step/tensorflow/mnist-distributed/job.yml
@@ -7,7 +7,7 @@ command: >-
 inputs:
   epochs: 1
   model_dir: outputs/keras-model
-environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda12@latest
 compute: azureml:gpu-cluster
 resources:
   instance_count: 2

diff --git a/cli/jobs/single-step/tensorflow/mnist/job.yml b/cli/jobs/single-step/tensorflow/mnist/job.yml
@@ -2,7 +2,7 @@ $schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
 code: src
 command: >-
   python train.py
-environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda12@latest
 compute: azureml:gpu-cluster
 display_name: tensorflow-mnist-example
 experiment_name: tensorflow-mnist-example

diff --git a/sdk/python/jobs/single-step/tensorflow/mnist-distributed/tensorflow-mnist-distributed.ipynb b/sdk/python/jobs/single-step/tensorflow/mnist-distributed/tensorflow-mnist-distributed.ipynb
@@ -130,7 +130,7 @@
     "    code=\"./src\",  # local path where the code is stored\n",
     "    command=\"python main.py --epochs ${{inputs.epochs}} --model-dir ${{inputs.model_dir}}\",\n",
     "    inputs={\"epochs\": 1, \"model_dir\": \"outputs/keras-model\"},\n",
-    "    environment=\"AzureML-tensorflow-2.16-cuda11@latest\",\n",
+    "    environment=\"AzureML-tensorflow-2.16-cuda12@latest\",\n",
     "    compute=\"cpu-cluster\",\n",
     "    instance_count=2,\n",
     "    # distribution = {\"type\": \"mpi\", \"process_count_per_instance\": 1},\n",

diff --git a/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb b/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb
@@ -128,7 +128,7 @@
     "job = command(\n",
     "    code=\"./src\",  # local path where the code is stored\n",
     "    command=\"python main.py\",\n",
-    "    environment=\"AzureML-tensorflow-2.16-cuda11@latest\",\n",
+    "    environment=\"AzureML-tensorflow-2.16-cuda12@latest\",\n",
     "    display_name=\"tensorflow-mnist-example\"\n",
     "    # experiment_name: tensorflow-mnist-example\n",
     "    # description: Train a basic neural network with TensorFlow on the MNIST dataset.\n",