Search code examples
python-3.xpytorchtorchvision

How to check which pytorch version fits torchvision

I am trying to clone and run this repository: https://github.com/switchablenorms/CelebAMask-HQ

The demo runs with PyTorch 0.4.1, and I am trying to find the corresponding version, how can I find it?

Solution

  • The corresponding torchvision version for 0.4.1 is 0.2.1.

    The easiest way is to look it up in the previous versions section. Only if you couldn't find it, you can have a look at the torchvision release data and pytorch's version. There you can find which version, got release with which version!