Skip to content

Commit b0488d7

Browse files
samaidbrycelelbach
authored andcommitted
Added dockerfile and restructured directories accordingly
1 parent aa2b7cc commit b0488d7

22 files changed

+113
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
dockerfile
Lines changed: 83 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
x-config:
2+
dockerfile: &dockerfile tutorials/nvmath-python/brev/dockerfile
3+
image: &image ghcr.io/nvidia/nvmath-python-tutorial:brev-reorg-latest
4+
working-dir: &working-dir /accelerated-computing-hub/tutorials/nvmath-python/notebooks
5+
large: &large true
6+
7+
services:
8+
base:
9+
build:
10+
context: ../../..
11+
dockerfile: *dockerfile
12+
args:
13+
GIT_BRANCH_NAME: ${GIT_BRANCH_NAME:-HEAD}
14+
image: *image
15+
pull_policy: missing
16+
volumes:
17+
- accelerated-computing-hub:/accelerated-computing-hub
18+
environment:
19+
- BREV_ENV_ID
20+
- GIT_BRANCH_NAME
21+
user: root
22+
working_dir: *working-dir
23+
entrypoint: ["/opt/update-git-branch.bash"]
24+
restart: "no"
25+
jupyter:
26+
image: *image
27+
depends_on:
28+
base:
29+
condition: service_completed_successfully
30+
privileged: true
31+
ulimits:
32+
memlock: -1
33+
stack: 67108864
34+
shm_size: 1g
35+
deploy:
36+
resources:
37+
reservations:
38+
devices:
39+
- driver: nvidia
40+
count: all
41+
capabilities: [gpu]
42+
ports:
43+
- "0.0.0.0:8888:8888" # JupyterLab
44+
volumes_from:
45+
- base
46+
environment:
47+
- BREV_ENV_ID
48+
- GIT_BRANCH_NAME
49+
user: root
50+
working_dir: *working-dir
51+
restart: unless-stopped
52+
nsight:
53+
image: nvcr.io/nvidia/devtools/nsight-streamer-nsys:2025.3.1
54+
depends_on:
55+
base:
56+
condition: service_completed_successfully
57+
pull_policy: missing
58+
ulimits:
59+
memlock: -1
60+
stack: 67108864
61+
shm_size: 1g
62+
deploy:
63+
resources:
64+
reservations:
65+
devices:
66+
- driver: nvidia
67+
count: all
68+
capabilities: [gpu]
69+
ports:
70+
- "0.0.0.0:8080:8080" # HTTP
71+
- "0.0.0.0:3478:3478" # TURN
72+
volumes_from:
73+
- base
74+
environment:
75+
- BREV_ENV_ID
76+
- GIT_BRANCH_NAME
77+
user: root
78+
working_dir: *working-dir
79+
entrypoint: ["/accelerated-computing-hub/brev/nsight-start.bash"]
80+
restart: unless-stopped
81+
82+
volumes:
83+
accelerated-computing-hub:
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
# Pin NVIDIA library versions to match PyTorch requirements
2+
nvidia-cublas-cu12==12.8.4.1
3+
nvidia-nvshmem-cu12==3.3.20
4+
nvidia-cuda-nvcc-cu12==12.8.*
5+
nvidia-cuda-nvrtc-cu12==12.8.*
6+
7+
# CUDA
8+
cuda-python == 12.9.1 # cuda.{core, bindings}, numba.cuda
9+
10+
# Scientific
11+
numpy
12+
scipy
13+
ssgetpy
14+
cupy-cuda12x
15+
16+
nvmath-python[cu12,dx,cpu,cu12-distributed]
17+
18+
# Visualization
19+
matplotlib
20+
21+
# Jupyter
22+
jupyterlab
23+
jupyterlab-nvidia-nsight
24+
ipywidgets
25+
ipykernel
26+
27+
# MPI
28+
mpi4py
29+

0 commit comments

Comments
 (0)