Torchvision pip. To do this, run: poetry add torchvision Using version ^0.