Skip to content

Commit 56047ab

Browse files
committed
Add GPU and JAX hardware details to status page
- Add nvidia-smi output to show GPU availability - Add JAX backend check to confirm GPU usage - Matches format used in lecture-python.myst
1 parent e2939c2 commit 56047ab

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

lectures/status.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,18 @@ and the following package versions
3131
```{code-cell} ipython
3232
:tags: [hide-output]
3333
!conda list
34+
```
35+
36+
This lecture series has access to the following GPU
37+
38+
```{code-cell} ipython
39+
!nvidia-smi
40+
```
41+
42+
You can check the backend used by JAX using:
43+
44+
```{code-cell} ipython3
45+
import jax
46+
# Check if JAX is using GPU
47+
print(f"JAX backend: {jax.devices()[0].platform}")
3448
```

0 commit comments

Comments
 (0)