File size: 351 Bytes
5fa1a76 |
1 2 3 4 5 6 7 |
With the mps device set, you can: train larger networks or batch sizes locally reduce data retrieval latency because the GPU's unified memory architecture allows direct access to the full memory store reduce costs because you don't need to train on cloud-based GPUs or add additional local GPUs Get started by making sure you have PyTorch installed. |