-
Notifications
You must be signed in to change notification settings - Fork 9
/
Dockerfile
27 lines (17 loc) · 1.04 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
FROM rapidsai/rapidsai:0.11-cuda10.1-runtime-ubuntu18.04
ENV CONDA_ENV rapids
RUN source activate $CONDA_ENV && \
apt-get update && \
apt-get install -y screen unzip git vim htop font-manager && \
rm -rf /var/lib/apt/*
RUN source activate $CONDA_ENV && jupyter labextension install @jupyter-widgets/jupyterlab-manager
RUN source activate $CONDA_ENV && conda install -y -c conda-forge ipyvolume && conda clean -yac *
RUN source activate $CONDA_ENV && jupyter labextension install ipyvolume
RUN source activate $CONDA_ENV && conda install -c conda-forge python-graphviz && conda clean -yac *
RUN git clone https://github.com/miroenev/rapids
# enables demo of ETL with RAPIDS and model building with DL-framework [ optional extension ]
RUN source activate $CONDA_ENV && conda install -y -c pytorch pytorch
EXPOSE 8888
WORKDIR /
# the runtime RAPIDS container automatically launches a Jupyter Lab instances on port 8888
# CMD ["bash", "-c", "source activate $CONDA_ENV && jupyter lab --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token=''"]