Skip to content

Commit b413ea0

Browse files
author
Azure Pipelines
committed
publish [GPU]: lightning_examples/mnist-tpu-training
1 parent 7382f6a commit b413ea0

File tree

2 files changed

+31
-0
lines changed

2 files changed

+31
-0
lines changed
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:bdf9cb27687704ad71dcb393aed095929f1130fb58ee53ec8fd831656ae02fe5
3+
size 14571
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
title: TPU training with PyTorch Lightning
2+
author: Lightning.ai
3+
created: 2020-12-21
4+
updated: 2023-05-15
5+
license: CC BY-SA
6+
build: 0
7+
tags:
8+
- Image
9+
description: In this notebook, we'll train a model on TPUs. Updating one Trainer flag
10+
is all you need for that. The most up to documentation related to TPU training can
11+
be found [here](https://lightning.ai/docs/pytorch/stable/accelerators/tpu.html).
12+
requirements:
13+
- torchvision
14+
- lightning>=2.0.0
15+
accelerator:
16+
- TPU
17+
environment:
18+
- setuptools==69.0.3
19+
- numpy==1.26.4
20+
- lightning==2.3.3
21+
- urllib3==2.2.2
22+
- torch==2.3.1
23+
- ipython==8.16.1
24+
- torchmetrics==1.2.1
25+
- pytorch-lightning==1.5.3
26+
- matplotlib==3.8.4
27+
- torchvision==0.18.1
28+
published: '2024-07-20T00:13:16.860265'

0 commit comments

Comments
 (0)