After you've cloned the repository, you need to install required packages.
Run the following command to install packages:
pip install -r requirements.txtInstall PyTorch libraries with the command from official web-site:
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118Then install the rest of the libraries with the command:
pip install -r requirements-cuda.txtpython3 -m tensorboard.main --logdir=lightning/lightning_logs/