24. Execution Statistics#

This table contains the latest execution statistics.

Document

Modified

Method

Run Time (s)

Status

aiyagari_jax

2024-04-11 21:57

cache

67.88

arellano

2024-04-11 21:57

cache

28.02

cake_eating_numerical

2024-04-11 21:58

cache

18.82

hopenhayn

2024-04-11 22:01

cache

217.76

ifp_egm

2024-04-11 22:04

cache

179.8

intro

2024-04-11 22:04

cache

1.17

inventory_dynamics

2024-04-11 22:05

cache

69.69

inventory_ssd

2024-04-11 22:30

cache

1456.09

jax_intro

2024-04-11 22:30

cache

24.87

job_search

2024-04-11 22:30

cache

12.75

kesten_processes

2024-04-11 22:31

cache

19.92

lucas_model

2024-04-11 22:31

cache

19.94

markov_asset

2024-04-11 22:31

cache

17.45

mle

2024-04-11 22:32

cache

15.02

newtons_method

2024-04-11 22:34

cache

158.98

opt_invest

2024-04-11 22:34

cache

18.02

opt_savings_1

2024-04-11 22:35

cache

34.09

opt_savings_2

2024-04-11 22:35

cache

17.48

overborrowing

2024-04-18 07:24

cache

46.23

short_path

2024-04-11 22:36

cache

6.54

status

2024-04-11 22:36

cache

3.74

troubleshooting

2024-04-11 22:04

cache

1.17

wealth_dynamics

2024-04-11 22:37

cache

23.75

zreferences

2024-04-11 22:04

cache

1.17

These lectures are built on linux instances through github actions and amazon web services (aws) to enable access to a gpu. These lectures are built on a p3.2xlarge that has access to 8 vcpu's, a V100 NVIDIA Tesla GPU, and 61 Gb of memory.

You can check the backend used by JAX using:

import jax
# Check if JAX is using GPU
print(f"JAX backend: {jax.devices()[0].platform}")
JAX backend: gpu

and the hardware we are running on:

!nvidia-smi
/opt/conda/envs/quantecon/lib/python3.11/pty.py:89: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
  pid, fd = os.forkpty()
Thu Apr 11 22:36:38 2024       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.182.03   Driver Version: 470.182.03   CUDA Version: 12.3     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Tesla V100-SXM2...  Off  | 00000000:00:1E.0 Off |                    0 |
| N/A   45C    P0    44W / 300W |    310MiB / 16160MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
+-----------------------------------------------------------------------------+