GTX16xx用户的大坑,基本上每个GTX16xx用户使用YOLO系列算法,都会遇到这些问题
这个没办法,唯一的解决办法是使用cuda10.2的配置
https://download.pytorch.org/whl/torch/
解决方法, pip uninstall torchvision
重新安装对应版本
pip install torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html