MAINT: Update remaining workflows for new docker + anaconda + cuda (#331)

* ENH: Upgrade docker with anaconda==2023.02

* update matplotlib project='3d'

* change matplotlib syntax

* update env

* remove jax install from github actions

* TST: full build

* upgrade to cuda12 + jax

* update all workflows and update environment

* remove upgrade script

* fix syntax

* re-enable cache

Author: mmcky

.github/workflows/cache.yml

@@ -27,17 +27,15 @@ jobs:
     needs: deploy-runner
     runs-on: [self-hosted, cml-gpu]
     container:
-      image: docker://mmcky/quantecon-lecture-python:py39-anaconda-2022-10-jb-0.15.1
+      image: docker://mmcky/quantecon-lecture-python:cuda-12.1.0-anaconda-2023-03-py310
       options: --gpus all
     steps:
       - uses: actions/checkout@v3
         with:
           ref: ${{ github.event.pull_request.head.sha }}
-      - name: Install Jax and Upgrade CUDA
+      - name: Check nvidia drivers
         shell: bash -l {0}
         run: |
-          pip install --upgrade "jax[cuda]==0.4.2" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-          pip install --upgrade "numpyro[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
           nvidia-smi
       - name: Build HTML
         shell: bash -l {0}

.github/workflows/ci.yml

@@ -40,13 +40,13 @@ jobs:
       - name: Display Pip Versions
         shell: bash -l {0}
         run: pip list
-      # - name: Download "build" folder (cache)
-      #   uses: dawidd6/action-download-artifact@v2
-      #   with:
-      #     workflow: cache.yml
-      #     branch: main
-      #     name: build-cache
-      #     path: _build
+      - name: Download "build" folder (cache)
+        uses: dawidd6/action-download-artifact@v2
+        with:
+          workflow: cache.yml
+          branch: main
+          name: build-cache
+          path: _build
       # Build Assets (Download Notebooks and PDF via LaTeX)
       - name: Build Download Notebooks (sphinx-tojupyter)
         shell: bash -l {0}

.github/workflows/execution.yml

@@ -11,7 +11,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"]
-        python-version: ["3.9"]
+        python-version: ["3.10"]
     steps:
       - name: Checkout
         uses: actions/checkout@v2
@@ -54,7 +54,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["macos-latest"]
-        python-version: ["3.8"]
+        python-version: ["3.10"]
     steps:
       - name: Checkout
         uses: actions/checkout@v2
@@ -86,7 +86,7 @@ jobs:
   #     fail-fast: false
   #     matrix:
   #       os: ["windows-latest"]
-  #       python-version: ["3.8"]
+  #       python-version: ["3.10"]
   #   steps:
   #     - name: Checkout
   #       uses: actions/checkout@v2

.github/workflows/linkcheck.yml

@@ -13,7 +13,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"]
-        python-version: ["3.9"]
+        python-version: ["3.10"]
     steps:
       - name: Checkout
         uses: actions/checkout@v2
@@ -23,7 +23,7 @@ jobs:
           auto-update-conda: true
           auto-activate-base: true
           miniconda-version: 'latest'
-          python-version: 3.9
+          python-version: '3.10'
           environment-file: environment.yml
           activate-environment: quantecon
       - name: Download "build" folder (cache)

.github/workflows/publish.yml

@@ -28,19 +28,17 @@ jobs:
     needs: deploy-runner
     runs-on: [self-hosted, cml-gpu]
     container:
-      image: docker://mmcky/quantecon-lecture-python:py39-anaconda-2022-10-jb-0.15.1
+      image: docker://mmcky/quantecon-lecture-python:cuda-12.1.0-anaconda-2023-03-py310
       options: --gpus all
     steps:
       - name: Checkout
         uses: actions/checkout@v3
       - name: Install Git (required to commit notebooks)
         shell: bash -l {0}
         run: apt-get install -y git
-      - name: Install Jax and Upgrade CUDA
+      - name: Check nvidia drivers
         shell: bash -l {0}
         run: |
-          pip install --upgrade "jax[cuda]==0.4.2" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-          pip install --upgrade "numpyro[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
           nvidia-smi
       - name: Display Conda Environment Versions
         shell: bash -l {0}

environment.yml

@@ -2,8 +2,8 @@ name: quantecon
 channels:
   - default
 dependencies:
-  - python=3.9
-  - anaconda=2022.10
+  - python=3.10
+  - anaconda=2023.03
   - pip
   - pip:
     - jupyter-book==0.15.1