Pytorch-lightning CPU-only installation #9325
Answered
by
awaelchli
sivakhno
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
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 |
Beta Was this translation helpful? Give feedback.
Answered by
awaelchli
Sep 4, 2021
Replies: 1 comment 3 replies
-
Hey, if you install pytorch first (cpu only) and then Lightning it will just use that version. |
Beta Was this translation helpful? Give feedback.
3 replies
Answer selected by
sivakhno
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hey, if you install pytorch first (cpu only) and then Lightning it will just use that version.