diff --git a/notebook_instance_basic/main.tf b/notebook_instance_basic/main.tf index 1125e3de..9f8dd68f 100644 --- a/notebook_instance_basic/main.tf +++ b/notebook_instance_basic/main.tf @@ -3,7 +3,7 @@ resource "google_notebooks_instance" "instance" { location = "us-west1-a" machine_type = "e2-medium" vm_image { - project = "deeplearning-platform-release" - image_family = "pytorch-latest-cu124" + project = "cloud-notebooks-managed" + image_family = "workbench-instances" } } diff --git a/notebook_instance_basic_gpu/main.tf b/notebook_instance_basic_gpu/main.tf index 7173c64a..65f48dd3 100644 --- a/notebook_instance_basic_gpu/main.tf +++ b/notebook_instance_basic_gpu/main.tf @@ -9,7 +9,7 @@ resource "google_notebooks_instance" "instance" { core_count = 1 } vm_image { - project = "deeplearning-platform-release" - image_family = "pytorch-latest-cu124" + project = "cloud-notebooks-managed" + image_family = "workbench-instances" } } diff --git a/notebook_instance_basic_stopped/main.tf b/notebook_instance_basic_stopped/main.tf index 7dc31b72..f68d34a3 100644 --- a/notebook_instance_basic_stopped/main.tf +++ b/notebook_instance_basic_stopped/main.tf @@ -3,8 +3,8 @@ resource "google_notebooks_instance" "instance" { location = "us-west1-a" machine_type = "e2-medium" vm_image { - project = "deeplearning-platform-release" - image_family = "pytorch-latest-cu124" + project = "cloud-notebooks-managed" + image_family = "workbench-instances" } desired_state = "STOPPED" }