Install GPU driver
Install Docker
https://medium.com/@linhlinhle997/how-to-install-docker-and-nvidia-docker-2-0-on-ubuntu-18-04-da3eac6ec494https://medium.com/@linhlinhle997/how-to-install-docker-and-nvidia-docker-2-0-on-ubuntu-18-04-da3eac6ec494Install CUDA (not required by Docker)
Command
nvidia-docker run -v ~/docker:/workspace -it --name=screenshot --net=host pytorch/pytorch