- 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 开始)
2023-03-03
.is_available(): 检测 cuda 是否可用, 返回 True / False.device_count(): 返回 GPU 数量 (即使没有 GPU 也会返回 0).get_device_name(0): 返回 GPU 名 (索引从 0 开始)