Skip to content

Commit 37ad72c

Browse files
committed
revert to use in-lecture option using kaleido to install chrome
1 parent 8e8526f commit 37ad72c

File tree

2 files changed

+18
-15
lines changed

2 files changed

+18
-15
lines changed

.github/workflows/ci.yml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,12 @@ jobs:
2626
pip install --upgrade "jax[cuda12-local]==0.6.2"
2727
pip install numpyro pyro-ppl
2828
python scripts/test-jax-install.py
29-
- name: Install Chrome
30-
shell: bash -l {0}
31-
run: |
32-
wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb
33-
sudo apt install -y ./google-chrome-stable_current_amd64.deb
34-
google-chrome --version
29+
# - name: Install Chrome
30+
# shell: bash -l {0}
31+
# run: |
32+
# wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb
33+
# sudo apt install -y ./google-chrome-stable_current_amd64.deb
34+
# google-chrome --version
3535
- name: Check nvidia Drivers
3636
shell: bash -l {0}
3737
run: nvidia-smi

lectures/back_prop.md

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,17 @@ In addition to what's included in base Anaconda, we need to install the followin
2323
2424
!pip install -U kaleido plotly
2525
!conda install -y -c plotly plotly-orca
26+
27+
# kaleido needs chrome to build images
28+
import kaleido
29+
kaleido.get_chrome_sync()
30+
```
31+
32+
```{note}
33+
If you are running this on Google Colab the above cell will
34+
present an error. This is because Google Colab doesn't use Anaconda to manage
35+
the Python packages. However this lecture will still execute as Google Colab
36+
has `plotly` installed.
2637
```
2738

2839
We also need to install JAX to run this lecture
@@ -35,15 +46,7 @@ We also need to install JAX to run this lecture
3546

3647
```{code-cell} ipython3
3748
import jax
38-
## to check that gpu is activated in environment
39-
print(f"JAX backend: {jax.devices()[0].platform}")
40-
```
41-
42-
```{note}
43-
If you are running this on Google Colab the above cell will
44-
present an error. This is because Google Colab doesn't use Anaconda to manage
45-
the Python packages. However this lecture will still execute as Google Colab
46-
has `plotly` installed.
49+
print(f"JAX backend: {jax.devices()[0].platform}") # to check that gpu is activated in environment
4750
```
4851

4952
## Overview

0 commit comments

Comments
 (0)