lightning/docker
Jirka Borovec 6673fc9a0b
fix docker builds (#2383)
2020-06-27 08:49:19 -04:00
..
Dockerfile fix docker builds (#2383) 2020-06-27 08:49:19 -04:00
README.md fix docker builds (#2383) 2020-06-27 08:49:19 -04:00

README.md

Builds

You can build it on your own, note it takes lots of time, be prepared.

git clone <git-repository>
docker image build -t pytorch-lightning:latest -f docker/Dockerfile .

or with specific arguments

git clone <git-repository>
docker image build \
    -t pytorch-lightning:py36 \
    -f docker/Dockerfile \
    --build-arg PYTHON_VERSION=3.6 \
    --build-arg PYTORCH_VERSION=1.3 \
    .

To run your docker use

docker image list
docker run --rm -it pytorch-lightning:latest bash

and if you do not need it anymore, just clean it:

docker image list
docker image rm pytorch-lightning:latest