ENH: Upgrade docker with anaconda==2023.03 (#327)

* 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

Author: mmcky

.github/workflows/ci.yml

@@ -24,19 +24,12 @@ 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 }}
-      # Install Hardware Dependant Libraries
-      - name: Install Jax and Upgrade CUDA
-        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
       # Check nvidia drivers
       - name: nvidia Drivers
         shell: bash -l {0}
@@ -47,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}

environment.yml

@@ -11,9 +11,9 @@ dependencies:
     - sphinx-tojupyter==0.3.0
     - sphinxext-rediraffe==0.2.7
     - sphinx-exercise==0.4.1
-    - ghp-import==1.1.0
-    - sphinxcontrib-youtube==1.1.0
-    - sphinx-togglebutton==0.3.1
+    - ghp-import==2.1.0
+    - sphinxcontrib-youtube==1.2.0
+    - sphinx-togglebutton==0.3.2
     - arviz==0.13.0
     - kaleido
     # Docker Requirements

lectures/linear_algebra.md

@@ -288,8 +288,7 @@ The span is a two-dimensional plane passing through these two points and the ori
 ---
 tags: [output_scroll]
 ---
-fig = plt.figure(figsize=(10, 8))
-ax = fig.gca(projection='3d')
+ax = plt.figure(figsize=(10, 8)).add_subplot(projection='3d')
 
 x_min, x_max = -5, 5
 y_min, y_max = -5, 5

lectures/lln_clt.md

@@ -478,8 +478,7 @@ S = Z.cumsum(axis=1)
 Y = (1 / np.sqrt(ns)) * S
 
 # Plot
-fig = plt.figure(figsize = (10, 6))
-ax = fig.gca(projection='3d')
+ax = plt.figure(figsize = (10, 6)).add_subplot(projection='3d')
 
 a, b = -3, 3
 gs = 100