torch

• install:: pip3 install torch
• import:: import torch
• github:: pytorch/pytorch
• docs:: PyTorch

---

cuda §

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