MNIST example with PyTorch Lightning
pip install -r requirements.txt
If you encounter warning messages that your current GPU is not compatible with the current PyTorch installation, uninstall the pytorch installations by
pip uninstall torch torchvision torchaudio
and then refer to https://pytorch.org/ to get the required CUDA version suitable to your system.