Table of contents
In this article, we’ll build a Docker container for Machine Learning (ML) development environment. This image is quite useful if you’re developing ML models or you need a pre-configured Jupyter notebook with some of the most useful libraries.
Recently we published an article Quick And Simple Introduction to Kubernetes Helm Charts in 10 minutes, where you can find instructions on how to use Helm to deploy this container to your Kubernetes cluster.
Update for 2020
- Upgraded to Python 3.6.
- Fixed a lots of build issues.
Last time we’ve created Docker container with Jupiter, Keras, Tensorflow, Pandas, Sklearn and Matplotlib. Suddenly, I understood, that I’ve missed OpenCV for Docker image and video manipulations. Well, I spent whole day preparing new image build. And in this article I’ll show you how to do it much faster using Anaconda official Docker Image.
There’re two ways to do that.
This process takes ~7 minutes to build the container of 3.11 Gb in size.
When I started playing with ML in 2018 Anaconda was a super fast and easiest way to create Docker container for ML experiments. It was much faster, then to compile OpenCV 3 for Ubuntu 16.04. Today it’s vice versa.
I’m using the same sources, but changing
Here how it looks like:
FROM continuumio/anaconda3 MAINTAINER "Andrei Maksimov" RUN apt-get update && apt-get install -y libgtk2.0-dev && \ rm -rf /var/lib/apt/lists/* RUN /opt/conda/bin/conda update -n base -c defaults conda && \ /opt/conda/bin/conda install python=3.6 && \ /opt/conda/bin/conda install anaconda-client && \ /opt/conda/bin/conda install jupyter -y && \ /opt/conda/bin/conda install --channel https://conda.anaconda.org/menpo opencv3 -y && \ /opt/conda/bin/conda install numpy pandas scikit-learn matplotlib seaborn pyyaml h5py keras -y && \ /opt/conda/bin/conda upgrade dask && \ pip install tensorflow imutils RUN ["mkdir", "notebooks"] COPY conf/.jupyter /root/.jupyter COPY run_jupyter.sh / # Jupyter and Tensorboard ports EXPOSE 8888 6006 # Store notebooks in this mounted directory VOLUME /notebooks CMD ["/run_jupyter.sh"]
As you can see, we’re installing just only libgtk2.0 for OpenCV support and all the other components like Terraform, Pandas, Scikit-learn, Matplotlib, Keras and others using conda package manager.
Now you have a working container and it’s time to start it. Create a folder inside your project’s folder where we’ll store all our Jupyter Noteboos with source code of our projects:
And start the container with the following command:
docker run -it -p 8888:8888 -p 6006:6006 \ -d -v $(pwd)/notebooks:/notebooks \ python_data_science_container:anaconda
It will start the container and expose Jupyter on port
8888 and Tensorflow Dashboard on port
6006 on your local computer or your server depending on where you’re executed this command.
If you don’t want to create and maintain your own container, please feel free to use my personal container:
docker run -it -p 8888:8888 -p 6006:6006 -d -v \ $(pwd)/notebooks:/notebooks amaksimov/python_data_science:anaconda
Installing Additional Packages
As soon as you’ve launched Jupyter, some packages may be missing for you and it’s OK. Feel free to to run the following command in a cell of your Jupyter notebook:
!pip install requests
Or for conda:
!conda install scipy
Hope, this article was helpful for you. If so, please like or repost it. See you soon!
Using Anaconda as a base image makes your Docker image heavy. I mean REALLY heavy.
docker images REPOSITORY TAG IMAGE ID CREATED SIZE amaksimov/python_data_science anaconda 7021f28dfba1 29 minutes ago 6.36GB amaksimov/python_data_science latest 3330c8eaec1c 2 hours ago 3.11GB
Installing all the components inside the Ubuntu 20.04 LTS container image including OpenCV 3 takes ~7 minutes, and final image ~3.11 Gb.
At the same time Anaconda3 container creation process takes x2 times longer and it gives you x2 times bigger image (~6.36 Gb). The building process is much more complicated, then it was in 2018, and it took me a while to update configuration to a working state.
We hope, you found this article helpful. If so, please, help us to spread it to the world!