A deep-learning docker image for Colab (use as local runtime) - two flavors

cupy.Dockerfile

FROM tiangolo/python-machine-learning:cuda9.1-cudnn7-devel-python3.6

# RUN conda install tensorflow-gpu
RUN conda install -y cudatoolkit=9.0
RUN conda install -y jupyter matplotlib pandas pillow seaborn tqdm
RUN pip install h5py wandb
RUN pip install cupy-cuda91==7.0.0b3
RUN pip install jupyter_http_over_ws
RUN jupyter serverextension enable --py jupyter_http_over_ws

# RUN mkdir -p /jupyter/data  # should be mounted instead

EXPOSE 8888

WORKDIR /jupyter

CMD ["jupyter", "notebook", "--allow-root", "--no-browser", "--ip=0.0.0.0", "--port=8888", "--NotebookApp.allow_origin='https://colab.research.google.com'", "--NotebookApp.port_retries=0"]

tf2.Dockerfile

FROM tensorflow/tensorflow:2.0.0rc0-gpu-py3-jupyter

RUN apt update && apt install -y git

# RUN conda install cudatoolkit=9.0  # this messes with tensorflow apparently
# RUN pip install cupy-cuda100==7.0.0b3  # doesn't work without cudatoolkit from conda apparently
# pip install keras  # not compatible with tensorflow 2 yet
RUN pip install h5py pandas pillow scipy seaborn tqdm wandb
# RUN pip install jupyter_http_over_ws  # already contained in base image
# RUN jupyter serverextension enable --py jupyter_http_over_ws  # already contained in base image

# RUN mkdir -p /jupyter/data  # should be mounted instead

EXPOSE 8888

WORKDIR /jupyter

CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --allow-root --no-browser --ip=0.0.0.0 --port=8888 --NotebookApp.allow_origin='https://colab.research.google.com' --NotebookApp.port_retries=0"]