torch.accelerator
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.accelerator.device_count |
是 |
- |
torch.accelerator.is_available |
是 |
- |
torch.accelerator.current_accelerator |
是 |
- |
torch.accelerator.set_device_index |
是 |
- |
torch.accelerator.set_device_idx |
是 |
- |
torch.accelerator.current_device_index |
是 |
- |
torch.accelerator.current_device_idx |
是 |
- |
torch.accelerator.set_stream |
是 |
- |
torch.accelerator.current_stream |
是 |
- |
torch.accelerator.synchronize |
是 |
- |
父主题： PyTorch 2.6.0版本