Stable 1.0 with cuda 10 successfully, but cannot pip3 install torchvision!


(dyw) #1

I installed pytorch stable 1.0, linux, pip, python 3.6, successfully. However, when I installed torchvision using pip3, it began downloading a cpu-version pytorch. I cancelled the installation.

How to install torchvision, while pytorch is already installed.
Waiting for help!


(dyw) #2

I find that ‘pip3 install torchvision-nightly’ do well!


#3

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.