Skip to content

Commit 53d48a8

Browse files
committed
Simplify GPU admonition text
1 parent 8833382 commit 53d48a8

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

lectures/_admonition/gpu.md

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,5 @@
11
```{admonition} GPU
22
:class: warning
33
4-
This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and targets JAX for GPU programming.
5-
6-
Free GPUs are available on Google Colab.
7-
To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.
8-
9-
Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support.
10-
If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
4+
This lecture is designed to run on a GPU. To use Google Colab's free GPUs, click the play icon top right, select Colab, and set the runtime to include a GPU. For local GPU setup, see the [JAX installation guide](https://github.com/google/jax).
115
```

0 commit comments

Comments
 (0)