diff --git a/cli/jobs/basics/hello-interactive.yml b/cli/jobs/basics/hello-interactive.yml index d8bdd7ff219..e953b971356 100644 --- 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 index 1b388760ab7..225066af32a 100644 --- 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 index 7ef1eae3b35..227b369c2bc 100644 --- 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 index 31c2b337056..6f370a57ef3 100644 --- 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 index 76726421acf..dc4ba1e66da 100644 --- 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 index a8c3b04e003..fd6b3df65fc 100644 --- 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 index 4c8cf4c12f2..30a1df32bd5 100644 --- 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",