Installing a smaller version of Pytorch using pip

[rllyryan]

[rllyryan]

Dear to whom it may concern,

I have tried to install Pytorch on my instance but found that it is too big of a file (at ~700MB). Is there a minimal version of Pytorch that I could install for the main purpose of functioning in the production line (requires a few functions)?

Below are the only imports that were used:

import torch
import torch.nn as nn
from torch.distributions import MultivariateNormal
from torch.distributions import Categorical

If there is, may I have a recommended approach to install the CPU and/or the GPU supported versions of Pytorch (with terminal lines)?

Thank you so much!

[rllyryan]

I realised that this is the wrong place to answer this question, please remove it.

I apologise.