Install Torchsummary Conda. pip If you installed Python by any of the PyTorch On Linux: con