Skip to content

Instantly share code, notes, and snippets.

@hamelsmu
Created April 30, 2021 00:16
Show Gist options
  • Save hamelsmu/1aac1ba72a52ede7b9f3e89f43c63f04 to your computer and use it in GitHub Desktop.
Save hamelsmu/1aac1ba72a52ede7b9f3e89f43c63f04 to your computer and use it in GitHub Desktop.
Quickly Spin Up a Deep Learning Rig on Google Cloud
export IMAGE_FAMILY="pytorch-latest-gpu"
export ZONE="us-west1-b"
export INSTANCE_NAME="dl-rig"
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" \
--custom-memory="64GB" \
--boot-disk-size="250GB" \
--tags "http-server,https-server" \
--custom-cpu 16
@hamelsmu
Copy link
Author

After creating your VM you will need to enable SSH by doing the following

gcloud compute ssh $INSTANCE_NAME

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment