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}

- 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}

- 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

---

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

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