Create Dockerfile

[justusschock]

Before submitting

• Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?
• If you made a notable change (that affects users), did you update the CHANGELOG?

What does this PR do?

Adds a docker file to

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[Borda]

Are older pytorch version compatible? I remember that a month ago even 1.4 supports cuda 10.1 and 9.2 only
There was also a comment that that from pt 1.5 they support cuda 10.2

So maybe add a test call if installed pytorch accesses cuda correctly so we don't produce useless images...

[justusschock]

This shouldn't matter, since pytorch also comes with precompiled cuda. I chose that, since this is the version that may be used, if somebody wants to compile custom stuff.

[tgaddair]

It will be important for Horovod and I believe Apex/AMP, as they both compile from CUDA source.

[justusschock]

docker build currently fails, since horovod guys have a hard torch vision dependency, that installs unrestricted torch vision (defaults to latest if not specified by us): horovod/horovod#1898

[williamFalcon]

@tgaddair

[tgaddair]

docker build currently fails, since horovod guys have a hard torch vision dependency, that installs unrestricted torch vision (defaults to latest if not specified by us): horovod/horovod#1898

Hey @justusschock @williamFalcon, I'll make a PR to remove the torchvision dep. But in the meantime, as I mentioned in the other thread, you should be able to change horovod[pytorch] to horovod in your requirements-extra.txt to unblock.

[Borda]

Hey @justusschock @williamFalcon, I'll make a PR to remove the torchvision dep. But in the meantime, as I mentioned in the other thread, you should be able to change horovod[pytorch] to horovod in your requirements-extra.txt to unblock.

we just did in #1573

[Borda]

failed to build:

Step 14/15 : RUN python -c "import pytorch_ligtning as pl; print(pl.__version__)"
 ---> Running in 5e7a518cec95
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch_ligtning'
The command '/bin/sh -c python -c "import pytorch_ligtning as pl; print(pl.__version__)"' returned a non-zero code: 1

[justusschock]

Just a small typo. Fixed this :)

[justusschock]

@williamFalcon can we merge this?