Skip to content

Commit 7386e39

Browse files
committed
forgot to add attribution to new PyTorch page
1 parent d34280a commit 7386e39

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)