PyTorch Docker

Docker Setup

• docker hub: https://hub.docker.com/r/pytorch/pytorch/tags

$ docker pull pytorch/pytorch:1.3-cuda10.1-cudnn7-runtime

create and start

$ docker container create --gpus '"device=1"' -t -i  --name pytorch_v1_3  pytorch/pytorch:1.3-cuda10.1-cudnn7-runtime
$ docker container start -a -i pytorch_v1_3

run

docker container run --gpus '"device=1"' -it --rm --name pytorch_v1_3 -v ~/work2:/work2 pytorch/pytorch:1.3-cuda10.1-cudnn7-runtime