docker-for-ML

Docker image build recipe with various machine learning libraries, including tensorflow, pytorch and opencv. This repository is tested on ubuntu 18.04 with NVIDIA Geforce 1080TI and NVIDIA RTX 2080TI.

What you need to do first

1. Install docker according to the docker docs.
2. Install nvidia-docker according to the nvidia-docker github.
3. Clone this repository and enter into the directory.

git clone https://github.com/hslyu/docker-for-ML
cd ./docker-for-ML

4. In the docker-for-ML directory, clone tensorflow github and opencv github.
5. Go to the tensorflow directory and checkout to the version you want to build.
6. Go to the opencv directory and checkout to the version you want to build.