First Test for Syntax Highlighting

How to get current available GPUs in tensorflow/pytorch?

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

import torch
torch.__version__
torch.version.cuda
torch.cuda.device_count()
torch.cuda.get_device_name(0)

Written on January 14, 2019