Simple base image https://hub.docker.com/repository/docker/blume/pytorch docker build -t blume/pytorch:latest -t blume/pytorch:{version} . docker push blume/pytorch