Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/dockers
History
Jeff Yang 90929fa433
Fix apt repo issue for docker (#3823) 
* fix docker repo issue

* docker

* docker

* docker

* no cudnn

* no cudnn

* try 16.04

Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>
 		2020-10-05 23:18:14 -04:00
..
base-cuda Fix apt repo issue for docker (#3823) 2020-10-05 23:18:14 -04:00
base-xla fix path in CI for release & python version in all dockers & duplicated badges (#3765) 2020-10-02 05:26:21 -04:00
conda fix path in CI for release & python version in all dockers & duplicated badges (#3765) 2020-10-02 05:26:21 -04:00
tpu-tests fix path in CI for release & python version in all dockers & duplicated badges (#3765) 2020-10-02 05:26:21 -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