We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f1510b commit c23a7c0Copy full SHA for c23a7c0
.github/workflows/ci.yml
@@ -23,7 +23,7 @@ jobs:
23
run: |
24
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
25
pip install --upgrade "jax[cuda12-local]"
26
- pip install numpyro
+ pip install numpyro pyro-ppl
27
python scripts/test-jax-install.py
28
- name: Check nvidia Drivers
29
shell: bash -l {0}
0 commit comments