lightning/dockers
Jirka Borovec ab508dae0c
run TPU tests with multiple versions (#3024)
* rename

* multi build

* multi build

* copy

* copy

* copy

* copy

* copy

* copy

* clean

* note

* docker

* formatting

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-09-30 08:36:02 -04:00
..
base-cuda fix PT version in CUDA docker images (#3739) 2020-09-30 08:33:22 -04:00
base-xla run TPU tests with multiple versions (#3024) 2020-09-30 08:36:02 -04:00
conda Lightning docker image based on base-cuda (#3637) 2020-09-24 23:14:15 +02:00
tpu-tests run TPU tests with multiple versions (#3024) 2020-09-30 08:36:02 -04:00
README.md test dockers & add AMP in pt-1.6 (#1584) 2020-07-31 08:23:13 -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 dockers/conda/Dockerfile .

or with specific arguments

git clone <git-repository>
docker image build \
    -t pytorch-lightning:py3.8 \
    -f dockers/conda/Dockerfile \
    --build-arg PYTHON_VERSION=3.8 \
    --build-arg PYTORCH_VERSION=1.4 \
    .

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