Set up Pytorch Env with Docker within 1 Minute

  Aug 16, 2018      2m
   

Easily set up Pytorch env with only one command: $ docker-compose up. That's all! docker-compose.yml is at this repository.

Set up Pytorch Env with Docker within 1 Minute

This docker image gives the environment:

  • Python 3.5
  • Latest Pytorch framework
  • GPU supported
  • Useful libraries: numpy, matplotlib, opencv, ffmpeg
  • Jupyter Lab (it'll be extreamly helpful if your machine is a server)

Setup step-by-step:

$ git clone https://github.com/minhng92/pytorch-docker-jupyter
$ cd pytorch-docker-jupyter
$ mkdir home workspace
$ docker build . -t py35_gpu
$ docker-compose up -d

That's it. Go to localhost:10010 with a browser and enjoy your new Pytorch environment.

Github: https://github.com/minhng92/pytorch-docker-jupyter

Docker Jupyter


Want more docker compose?