GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision. We recommend Anaconda as Python package management system. Please refer to pytorch.org for the detail of PyTorch (torch) installation. The following is the corresponding torchvision versions and supported Python versions. We don't officially support building from source using pip, but if you do, you'll need to use the --no-build-isolation flag.