lightning

History

Jeff Yang 8be79a9a96 stable, dev PyTorch in Dockerfile and conda gh actions (#3074 ) * dockerfile and actions file * dockerfile and actions file * added pytorch conda cpu nightly * added pytorch conda cpu nightly * recopy base reqs * gh action `include` torch nightly * add pytorch nightly & conda gh badge * rebase * fix horovod * proposal refactor * Update .github/workflows/ci_pt-conda.yml Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * Update .github/workflows/ci_pt-conda.yml Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * update * update * fix cmd * filled && * fix * add -y * torchvision >0.7 allowed * explicitly install torchvision * use HOROVOD_GPU_OPERATIONS env variable * CI * skip 1.7 * table Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>		2020-09-17 20:30:39 +02:00
..
base-cuda	stable, dev PyTorch in Dockerfile and conda gh actions (#3074 )	2020-09-17 20:30:39 +02:00
base-xla	add CI for building dockers (#3383 )	2020-09-10 18:38:29 -04:00
conda	stable, dev PyTorch in Dockerfile and conda gh actions (#3074 )	2020-09-17 20:30:39 +02:00
tpu-tests	integrate with CircleCI (#2486 )	2020-07-23 12:13:10 -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