import tensorflow as tf
import torch
if __name__ == '__main__':
gpu_name = torch.cuda.get_device_name(0)
print(gpu_name)