Krischal's Digital Garden

Crossplatform code of getting default accerator in pytorch

1 min read

import torch
device = torch.accelerator.current_accelerator() or torch.device("cpu")

Graph View