how do i find out which pytorch version will work with torchvision

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!