Skip to content

Commit 01412e3

Browse files
committed
update GPU warning
1 parent af6b8e7 commit 01412e3

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

lectures/kesten_processes.md

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,16 +19,15 @@ kernelspec:
1919

2020
# Kesten Processes and Firm Dynamics
2121

22-
````{admonition} GPU Warning
22+
```{admonition} GPU Warning
2323
:class: dropdown, warning
2424
25-
This lecture is built using [hardware](status:machine-details) that
26-
has access to a GPU and uses JAX for GPU programming. This means that
25+
This lecture is built using [hardware](status:machine-details) that has access to a GPU and uses JAX for GPU programming. As a result, the lecture will be slower when running on a machine without a GPU.
2726
28-
1. this lecture will be significantly slower when running on a machine **without CUDA support**.
29-
In this case, the code is well-suited to execution with Google Colab with runtime type set to GPU. [Click here](https://colab.research.google.com/github/QuantEcon/lecture-python.notebooks/blob/master/kesten_processes.ipynb) to load this lecture on Colab;
30-
2. if you have CUDA support for your machine, you should follow the [instructions](https://github.com/google/jax#installation) for JAX with GPU support.
31-
````
27+
Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.
28+
29+
Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax#pip-installation-gpu-cuda) for installing JAX with GPU support.
30+
```
3231

3332
```{index} single: Linear State Space Models
3433
```

0 commit comments

Comments
 (0)