【yolov5问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘
cuda的版本和torch的版本不对应,把torch的版本和cuda的版本都降下来pipinstalltorch==1.9.0+cu111torchvision==0.10.0+cu111torchaudio===0.9.0-fhttps://download.pytorch.org/whl/torch_stable.html-ihttps://pypi.douban.com/simple