Public Repository

Last pushed: a year ago
Short Description
A Docker image of the most popular machine learning and deep learning libraries.
Full Description

AIA ML Standard Docker Image

Hardware Requirements:

Built only for Nvidia Cuda enabled GPU only.

Packages within Image:

Utilizes Ubuntu 16.04, CUDA 8.0, CuDNN 5.1. Image contains the following Machine Learning/Deep Learning tools:

  • Anaconda
  • Pytorch
  • Tensorflow
  • Keras
  • Theano
  • Sklearn

Docker Pull Instructions:

Python 2: docker pull shantanudev/aia-ml-image:py2_v1
Python 3: docker pull shantanudev/aia-ml-image:py3_v1

Instructions on Launching Notebook

nvidia-docker run -i -t -p 8888:8888 shantanudev/pwc-aia_ml_image /bin/bash -c "mkdir /opt/notebooks && /opt/conda/bin/jupyter notebook --notebook-dir=/opt/notebooks --ip='*' --port=8888 --no-browser"**

Note: Do not save anything to the docker because your files will be deleted as soon as the container is exited. You must mount the volumes from your local machine to the docker container in order to have your information saved. Source:

Docker Pull Command