Pytorch-lightning CPU-only installation
Hello all - just wanted to discuss a use-case with CPU vs GPU PL install. We do normal training on GPUs, but when deploying for prediction we use CPUs and would like to keep the Docker container size as small as possible. It't not clear if it's possible to install pytorch-lightning with CPU-only torch distribution, which is much smaller. Is there any possible equivalent to pip install pytorch-lightning[cpu] . Thanks for suggestions!
Hey, if you install pytorch first (cpu only) and then Lightning it will just use that version.