728x90
반응형
torchvision이 torch와 호환되지 않는 경우 발생하는 문제로 추측
1. torchvision 재설치 or torch version에 맞춰 torchvison 버전을 변경하거나
pip3 install torchvision==[$VERSION] -f https://download.pytorch.org/whl/torch_stable.html
ex) pip3 install torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
https://github.com/pytorch/pytorch/issues/42078
torchvision 0.7 has been released and is compatible with torch 1.6. It does not appear to be compatible with torch 1.5.
Note that installing torchvision via PyCharm may not result in an automatic upgrade of torch. Installing via pip3 did automatically upgrade torch however.
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
728x90
반응형