1. Installation

Prerequisites

This project has dependencies that require separate installation:

• PyTorch (version >= 1.10.1, < 2.0)
• Torchvision (version >= 0.11.2, < 0.12)
• TorchMetrics (version >= 0.7.2 < 0.8)

When possible, try to follow the recommended version range specified in parentheses.

You can install PyTorch and Torchvision together from here. And you install TorchMetrics from here.

After installing dependencies, simply execute the following command to install TorchAssistant from PyPI

pip install torchassistant