torch

2023-03-03


cuda

  • .is_available(): 检测 cuda 是否可用, 返回 True / False
  • .device_count(): 返回 GPU 数量 (即使没有 GPU 也会返回 0)
  • .get_device_name(0): 返回 GPU 名 (索引从 0 开始)