PyTorch for prediction only

Hello,
I already have a trained neural network, and now I use it for prediction.
I’d like to reduce the installation size due to deployment concerns.
Is there any partial torch packet with a smaller size that can be used to calculate prediction only (without training)?
Thanks,
Adi