Usage:
./check-instance.py
shows number of gpus pytorch has interface with and prints their names in stdout.
Install pytorch:
either choose Deep Learning GCP image here: GCP
or install pytorch with your favorite tool: Pytorch
example output:
Pytorch version: 1.0.0
Number of GPUs: 1
GPU 0: GeForce GTX 1080 Ti