How to specify preemptible GPU Deep Learning Virtual Machine on GCP

2
votes

I can't figure out how to specify preemptible GPU Deep Learning VM on GCP

This what I used:

export IMAGE_FAMILY="tf-latest-gpu"
export ZONE="europe-west4-a "
export INSTANCE_NAME="deeplearning"

gcloud compute instances create $INSTANCE_NAME \
  --zone=$ZONE \
  --image-family=$IMAGE_FAMILY \
  --image-project=deeplearning-platform-release \
  --maintenance-policy=TERMINATE \
  --accelerator='type=nvidia-tesla-v100,count=2' \
  --metadata='install-nvidia-driver=True'

Thank you!

1
google-cloud-platformdeep-learninggoogle-compute-engine

1 Answers

2
votes

You can create a preemptible Compute Engine instance with GPU by adding the --preemptible gcloud command option. As per your example, that would be:

export IMAGE_FAMILY="tf-latest-gpu"
export ZONE="europe-west4-a "
export INSTANCE_NAME="deeplearning"

gcloud compute instances create $INSTANCE_NAME \
  --zone=$ZONE \
  --image-family=$IMAGE_FAMILY \
  --image-project=deeplearning-platform-release \
  --maintenance-policy=TERMINATE \
  --accelerator type=nvidia-tesla-v100,count=2 \
  --metadata='install-nvidia-driver=True'
  --preemptible

See documentation here and here for more details on available options.