1import torch
2import torch.nn as nn
3import torch.nn.functional as F
4import torch.optim as optim

1print('cuda is available? ', torch.cuda.is_available())
2print('device_count: ', torch.cuda.device_count())
3print('current device: ', torch.cuda.current_device())
4print('device name: ', torch.cuda.get_device_name(0))

1# Determine supported device
2def get_device():
3    if torch.cuda.is_available():
4        device = torch.device('cuda:0') # or something else
5    else:
6        device = torch.device('cpu') # don't have GPU
7    return device

1def df_to_tensor(df):
2    device = get_device() # see other section
3    return torch.from_numpy(df.values).float().to(device)