File tree Expand file tree Collapse file tree 4 files changed +9
-30
lines changed
Expand file tree Collapse file tree 4 files changed +9
-30
lines changed Original file line number Diff line number Diff line change 1111 - uses : actions/checkout@v6
1212 with :
1313 ref : ${{ github.event.pull_request.head.sha }}
14- - name : Upgrade CUDANN
15- shell : bash -l {0}
16- run : |
17- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
18- sudo dpkg -i cuda-keyring_1.1-1_all.deb
19- sudo apt-get update
20- sudo apt-get -y install cudnn-cuda-12
2114 - name : Setup Anaconda
2215 uses : conda-incubator/setup-miniconda@v3
2316 with :
3023 - name : Install jax (and install checks for GPU)
3124 shell : bash -l {0}
3225 run : |
33- pip install -U "jax[cuda12 ]"
26+ pip install -U "jax[cuda13 ]"
3427 python --version
3528 python scripts/test-jax-install.py
3629 nvidia-smi
Original file line number Diff line number Diff line change 77 - uses : actions/checkout@v6
88 with :
99 ref : ${{ github.event.pull_request.head.sha }}
10- - name : Upgrade CUDANN
11- shell : bash -l {0}
12- run : |
13- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
14- sudo dpkg -i cuda-keyring_1.1-1_all.deb
15- sudo apt-get update
16- sudo apt-get -y install cudnn-cuda-12
1710 - name : Setup Anaconda
1811 uses : conda-incubator/setup-miniconda@v3
1912 with :
2619 - name : Install jax (and install checks for GPU)
2720 shell : bash -l {0}
2821 run : |
29- pip install -U "jax[cuda12-local ]"
22+ pip install -U "jax[cuda13 ]"
3023 python --version
3124 python scripts/test-jax-install.py
3225 nvidia-smi
Original file line number Diff line number Diff line change 1010 steps :
1111 - name : Checkout
1212 uses : actions/checkout@v6
13- - name : Upgrade CUDANN
14- shell : bash -l {0}
15- run : |
16- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
17- sudo dpkg -i cuda-keyring_1.1-1_all.deb
18- sudo apt-get update
19- sudo apt-get -y install cudnn-cuda-12
2013 - name : Setup Anaconda
2114 uses : conda-incubator/setup-miniconda@v3
2215 with :
2922 - name : Install jax (and install checks for GPU)
3023 shell : bash -l {0}
3124 run : |
32- pip install -U "jax[cuda12 ]"
25+ pip install -U "jax[cuda13 ]"
3326 python --version
3427 python scripts/test-jax-install.py
3528 nvidia-smi
Original file line number Diff line number Diff line change @@ -7,13 +7,13 @@ dependencies:
77 - pip
88 - pip :
99 - jupyter-book==1.0.4post1
10- - quantecon-book-theme==0.10.0
11- - sphinx-tojupyter==0.3.1
12- - sphinxext-rediraffe==0.2.7
13- - sphinx-exercise==1.0 .1
14- - sphinx-proof==0.2.1
10+ - quantecon-book-theme==0.13.2
11+ - sphinx-tojupyter==0.6.0
12+ - sphinxext-rediraffe==0.3.0
13+ - sphinx-exercise==1.2 .1
14+ - sphinx-proof==0.3.0
1515 - sphinxcontrib-youtube==1.4.1
1616 - sphinx-togglebutton==0.3.2
17- - sphinx-reredirects==0.1.4
17+ - sphinx-reredirects==1.0.0
1818
1919
You can’t perform that action at this time.
0 commit comments