Skip to content

Commit 01c9020

Browse files
authored
Merge pull request #213 from NYU-RTS/attribution_fix
forgot to add attribution to new PyTorch page
2 parents d34280a + 7386e39 commit 01c9020

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

docs/hpc/08_ml_ai_hpc/02_pytorch_intro.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,9 @@
11
# Single-GPU Training with PyTorch
22

3+
:::info
4+
This was adapted from [Princeton University Multi-GPU Training with PyTorch](https://github.com/PrincetonUniversity/multi_gpu_training)
5+
:::
6+
37
It is important to optimize your script for the single-GPU case before moving to multi-GPU training. This is because as you request more resources, your queue time increases. We also want to avoid wasting resources by running code that is not optimized.
48

59
Here we train a CNN on the MNIST dataset using a single GPU as an example. We profile the code and make performance improvements.

0 commit comments

Comments
 (0)