lightning

History

Jirka Borovec a6e7aa7796 allow using apex with any PT version (#2865 ) * wip * setup * type * name * wip * docs * imports * fix if * fix if * use_amp * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * todos Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2020-08-08 11:07:32 +02:00
..
conda	Docker: building XLA base image (#2494 )	2020-07-06 14:21:36 -04:00
cuda-extras	allow using apex with any PT version (#2865 )	2020-08-08 11:07:32 +02:00
tpu-extras	use XLA base image for TPU testing (#2536 )	2020-07-07 07:05:17 -04: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