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