It sounds like “pip3 install torchvision” started installing a cpu-only version of pytorch, which was not what you wanted because you have a gpu on your machine. If that is the case you should file an issue at our github page.