Skip to content

Commit 5ba7aa9

Browse files
committed
Move nvidia-smi check before JAX installation in ci.yml
1 parent 36b5772 commit 5ba7aa9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/ci.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,15 +28,15 @@ jobs:
2828
python-version: "3.13"
2929
environment-file: environment.yml
3030
activate-environment: quantecon
31+
- name: Check nvidia Drivers
32+
shell: bash -l {0}
33+
run: nvidia-smi
3134
- name: Install JAX and Numpyro
3235
shell: bash -l {0}
3336
run: |
3437
pip install -U "jax[cuda13]"
3538
pip install numpyro
3639
python scripts/test-jax-install.py
37-
- name: Check nvidia Drivers
38-
shell: bash -l {0}
39-
run: nvidia-smi
4040
- name: Display Conda Environment Versions
4141
shell: bash -l {0}
4242
run: conda list

0 commit comments

Comments
 (0)