tuoheng_algN/test/gpu/gputest.py

import tensorflow as tf
import torch

if __name__ == '__main__':
    gpu_name = torch.cuda.get_device_name(0)
    print(gpu_name)