lightning/docker/README.md

34 lines
648 B
Markdown

## Builds
You can build it on your own, note it takes lots of time, be prepared.
```bash
git clone <git-repository>
docker image build -t pytorch-lightning:latest -f docker/Dockerfile .
```
or with specific arguments
```bash
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
```bash
docker image list
docker run --rm -it pytorch-lightning:latest bash
```
and if you do not need it anymore, just clean it:
```bash
docker image list
docker image rm pytorch-lightning:latest
```