Pytorch Lightning updated their docker container

There is no old version, so we have to either update the project to use the newest version of lightning (not necessary of pytorch), or we write our own container and upload it to gitlab.