File tree Expand file tree Collapse file tree 3 files changed +14
-18
lines changed Expand file tree Collapse file tree 3 files changed +14
-18
lines changed Original file line number Diff line number Diff line change @@ -20,12 +20,11 @@ jobs:
2020 python-version : " 3.13"
2121 environment-file : environment.yml
2222 activate-environment : quantecon
23- - name : Install JAX, Numpyro, PyTorch
23+ - name : Install JAX and Numpyro
2424 shell : bash -l {0}
2525 run : |
26- pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
2726 pip install --upgrade "jax[cuda12-local]==0.6.2"
28- pip install numpyro
27+ pip install numpyro
2928 python scripts/test-jax-install.py
3029 - name : Check nvidia drivers
3130 shell : bash -l {0}
Original file line number Diff line number Diff line change @@ -28,13 +28,11 @@ jobs:
2828 python-version : " 3.13"
2929 environment-file : environment.yml
3030 activate-environment : quantecon
31- - name : Install JAX, Numpyro, PyTorch
31+ - name : Install JAX and Numpyro
3232 shell : bash -l {0}
3333 run : |
34- pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
35- pip install pyro-ppl
36- pip install --upgrade "jax[cuda12-local]==0.6.2"
37- pip install numpyro pyro-ppl
34+ pip install --upgrade "jax[cuda12-local]"
35+ pip install numpyro
3836 python scripts/test-jax-install.py
3937 - name : Check nvidia Drivers
4038 shell : bash -l {0}
@@ -45,13 +43,13 @@ jobs:
4543 - name : Display Pip Versions
4644 shell : bash -l {0}
4745 run : pip list
48- - name : Download "build" folder (cache)
49- uses : dawidd6/action-download-artifact@v11
50- with :
51- workflow : cache.yml
52- branch : main
53- name : build-cache
54- path : _build
46+ # - name: Download "build" folder (cache)
47+ # uses: dawidd6/action-download-artifact@v11
48+ # with:
49+ # workflow: cache.yml
50+ # branch: main
51+ # name: build-cache
52+ # path: _build
5553 # Build Assets (Download Notebooks and PDF via LaTeX)
5654 - name : Build Download Notebooks (sphinx-tojupyter)
5755 shell : bash -l {0}
Original file line number Diff line number Diff line change @@ -19,12 +19,11 @@ jobs:
1919 python-version : " 3.13"
2020 environment-file : environment.yml
2121 activate-environment : quantecon
22- - name : Install JAX, Numpyro, PyTorch
22+ - name : Install JAX and Numpyro
2323 shell : bash -l {0}
2424 run : |
25- pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
2625 pip install --upgrade "jax[cuda12-local]==0.6.2"
27- pip install numpyro
26+ pip install numpyro
2827 python scripts/test-jax-install.py
2928 - name : Check nvidia drivers
3029 shell : bash -l {0}
You can’t perform that action at this time.
0 commit comments