一开始以为是torch torchvision版本不一致,仔细检查其实不是这样,因为torchvision的cuda版本很隐蔽,如果torch是cu的,那么torchvision也必须,但是一般很难找。可以利用如下强制命令
pip install torchvision==0.8.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html
一开始以为是torch torchvision版本不一致,仔细检查其实不是这样,因为torchvision的cuda版本很隐蔽,如果torch是cu的,那么torchvision也必须,但是一般很难找。可以利用如下强制命令
pip install torchvision==0.8.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html