Fix install from pip

Since v0.3.4, pytorch_lightning released 1.0 which is incompatible. This release only limits lightning's version so that the install is compatible with the source code.