Skip to content

Instantly share code, notes, and snippets.

@ResidentMario
Created August 20, 2020 21:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ResidentMario/7505848f01262aaa80dc86cc759e17ab to your computer and use it in GitHub Desktop.
Save ResidentMario/7505848f01262aaa80dc86cc759e17ab to your computer and use it in GitHub Desktop.
FROM nvidia/cuda:10.1-base-ubuntu18.04
WORKDIR /spell
RUN apt-get update && \
apt-get install -y wget && rm -rf /var/lib/apt/lists/*
ENV CONDA_HOME=/root/anaconda/
RUN wget \
https://repo.anaconda.com/miniconda/Miniconda3-py37_4.8.3-Linux-x86_64.sh \
&& mkdir /root/.conda \
&& bash Miniconda3-py37_4.8.3-Linux-x86_64.sh -fbp $CONDA_HOME \
&& rm -f Miniconda3-py37_4.8.3-Linux-x86_64.sh
ENV PATH=/root/anaconda/bin:$PATH
# must be rapids=0.15 due to https://github.com/rapidsai/cudf/issues/5994
# must use numba=0.50.1 due to https://github.com/rapidsai/dask-cuda/pull/385
RUN conda install \
-c rapidsai-nightly -c nvidia -c conda-forge -c defaults \
rapids=0.15 cudatoolkit=10.1 numba==0.50.1 jupyterlab
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment