Academic GPU Jupyter Lab

This offering is currently in beta.

Documentation for Jupyter Lab can be found here: https://jupyterlab.readthedocs.io/en/latest/

The academic OpenOndemand offering at https://linuxlab.engr.wustl.edu offers GPU-enabled Jupyter Lab environments for instructors, TAs, and students in specific classes that require their use.

This application supports users running against a single nVIDIA RTX 4000 Ada GPU.

Resource Request

At job start, users may specify how much GPU RAM they wish to consume during their computations. Based on that choice, additional system resources are allocated. Depending on the class or assignment, you may have been directed which choice is the best fit for your project.

GPU RAM usage is not controlled by the scheduler. It is the responsibility of each user to make sure they use only that amount that has been requested.

CPU and System RAM usage is enforced.

20GB GPU RAM (Maximum)
- 25G of System RAM
- 4 CPU Cores
- This sets the GPU to exclusive access mode, and only the allocated job will be allowed access to the GPU
10GB GPU RAM
- 12G of System RAM
- 2 CPU Cores
4 GB or 2GB of GPU RAM
- 6G of System RAM
- 2 CPU Cores

Working Directory

You can direct Jupyter Lab to start within a specific directory - simply enter the full path in the text box provided. Your academic home directory is the default:

/home/warehouse/wustlkey

As an example, if you are in the class EIT300 and you are working on project 2 - you may have placed your files in

/home/warehouse/wustlkey/EIT300/project2

Hours

The scheduler puts a limit on the number of hours a job will run. You may enter a number of hours here up to the defined maximum.

Python Packages Available

Users can utilize “pip” to install more packages. Subsequent updates to this environment may or may not continue to support user installed packages.

The environment is built (as of 6/17/24) with:

channels:
- pytorch
- conda-forge
- nvidia
- defaults
dependencies:
- pip
- python=3.11
- pytorch
- pytorch-lightning
- torchvision
- pytorch-cuda=12.1
- numpy
- magma-cuda121
- torchaudio
- jupyter
- ipykernel
- websockify
- torchtriton

Package                   Version
------------------------- --------------
anyio                     4.4.0
argon2-cffi               23.1.0
argon2-cffi-bindings      21.2.0
arrow                     1.3.0
asttokens                 2.4.1
async-lru                 2.0.4
async-timeout             4.0.3
attrs                     23.2.0
Babel                     2.14.0
beautifulsoup4            4.12.3
bleach                    6.1.0
Brotli                    1.1.0
cached-property           1.5.2
certifi                   2024.6.2
cffi                      1.16.0
charset-normalizer        3.3.2
colorama                  0.4.6
comm                      0.2.2
cryptography              42.0.8
debugpy                   1.8.1
decorator                 5.1.1
defusedxml                0.7.1
entrypoints               0.4
exceptiongroup            1.2.0
executing                 2.0.1
fastjsonschema            2.20.0
filelock                  3.15.1
fqdn                      1.5.1
fsspec                    2024.6.0
gmpy2                     2.1.5
h11                       0.14.0
h2                        4.1.0
hpack                     4.0.0
httpcore                  1.0.5
httpx                     0.27.0
hyperframe                6.0.1
idna                      3.7
importlib_metadata        7.1.0
importlib_resources       6.4.0
ipykernel                 6.29.4
ipython                   8.25.0
ipywidgets                8.1.3
isoduration               20.11.0
jedi                      0.19.1
Jinja2                    3.1.4
json5                     0.9.25
jsonpointer               3.0.0
jsonschema                4.22.0
jsonschema-specifications 2023.12.1
jupyter                   1.0.0
jupyter_client            8.6.2
jupyter-console           6.6.3
jupyter_core              5.7.2
jupyter-events            0.10.0
jupyter-lsp               2.2.5
jupyter_server            2.14.1
jupyter_server_terminals  0.5.3
jupyterlab                4.2.2
jupyterlab_pygments       0.3.0
jupyterlab_server         2.27.2
jupyterlab_widgets        3.0.11
jwcrypto                  1.5.6
lightning-utilities       0.11.2
MarkupSafe                2.1.5
matplotlib-inline         0.1.7
mistune                   3.0.2
mpmath                    1.3.0
nbclient                  0.10.0
nbconvert                 7.16.4
nbformat                  5.10.4
nest_asyncio              1.6.0
networkx                  3.3
notebook                  7.2.1
notebook_shim             0.2.4
numpy                     2.0.0
overrides                 7.7.0
packaging                 24.1
pandocfilters             1.5.0
parso                     0.8.4
pexpect                   4.9.0
pickleshare               0.7.5
Pillow                    9.4.0
pip                       24.0
pkgutil_resolve_name      1.3.10
platformdirs              4.2.2
prometheus_client         0.20.0
prompt_toolkit            3.0.47
psutil                    5.9.8
ptyprocess                0.7.0
pure-eval                 0.2.2
pycparser                 2.22
Pygments                  2.18.0
PySocks                   1.7.1
python-dateutil           2.9.0
python-json-logger        2.0.7
pytorch-lightning         2.3.0
pytz                      2024.1
PyYAML                    6.0.1
pyzmq                     26.0.3
qtconsole                 5.5.2
QtPy                      2.4.1
redis                     5.0.6
referencing               0.35.1
requests                  2.32.3
rfc3339-validator         0.1.4
rfc3986-validator         0.1.1
rpds-py                   0.18.1
Send2Trash                1.8.3
setuptools                70.0.0
six                       1.16.0
sniffio                   1.3.1
soupsieve                 2.5
stack-data                0.6.2
sympy                     1.12.1
terminado                 0.18.1
tinycss2                  1.3.0
tomli                     2.0.1
torch                     2.3.1
torchaudio                2.3.1
torchmetrics              1.4.0.post0
torchvision               0.18.1
tornado                   6.4.1
tqdm                      4.66.4
traitlets                 5.14.3
triton                    2.3.1
types-python-dateutil     2.9.0.20240316
typing_extensions         4.12.2
typing-utils              0.1.0
uri-template              1.3.0
urllib3                   2.2.1
wcwidth                   0.2.13
webcolors                 24.6.0
webencodings              0.5.1
websocket-client          1.8.0
websockify                0.12.0
wheel                     0.43.0
widgetsnbextension        4.0.11
zipp                      3.19.2