Set up Pytorch Env with Docker within 1 Minute
Easily set up Pytorch env with only one command: $ docker-compose up. That's all! docker-compose.yml is at this repository.
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)
$ 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.
Want more docker compose?