• torch.cuda.is_available() cuda是否可用；
  • torch.cuda.device_count() 返回gpu数量；
  • torch.cuda.get_device_name(0) 返回gpu名字，设备索引默认从0开始；
  • torch.cuda.current_device()；
  • cuda是nvidia gpu的编程接口，opencl是amd gpu的编程接口