【yolov5问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘

【yolov5问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘_第1张图片
cuda的版本和torch的版本不对应,把torch的版本和cuda的版本都降下来

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

在这里插入图片描述
【yolov5问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘_第2张图片
【yolov5问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘_第3张图片

你可能感兴趣的:(YOLO,深度学习,pytorch,学习,笔记)