Skip to content

Commit 6c59739

Browse files
Install RAPIDS in the parabricks app (#240)
1 parent 9834807 commit 6c59739

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

src/workbench-jupyter-parabricks/Dockerfile

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,21 @@ USER ${NB_USER}
3636
ENV SHELL=/bin/bash
3737

3838
# Install Jupyter Lab using pip
39-
RUN pip3 install jupyterlab
39+
RUN pip3 install jupyterlab && \
40+
# Install NVIDIA Rapids
41+
pip3 install \
42+
--extra-index-url=https://pypi.nvidia.com \
43+
"cudf-cu12==25.8.*" \
44+
"dask-cudf-cu12==25.8.*" \
45+
"cuml-cu12==25.8.*" \
46+
"cugraph-cu12==25.8.*" \
47+
"nx-cugraph-cu12==25.8.*" \
48+
"cuxfilter-cu12==25.8.*" \
49+
"cucim-cu12==25.8.*" \
50+
"pylibraft-cu12==25.8.*" \
51+
"raft-dask-cu12==25.8.*" \
52+
"cuvs-cu12==25.8.*" \
53+
"nx-cugraph-cu12==25.8.*"
4054

4155
### BEGIN parabricks ###
4256

0 commit comments

Comments
 (0)