解决跑YOLOV5出现RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED的问题

解决跑YOLOV5出现RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED的问题

先确保torch与cuda版本是否匹配,可以在python命令下测试是否能用torch.cuda。如果返回True接着下面:
1.卸载原先的torch1.8.0版本(pip uninstall torch)
2.在conda环境下执行
pip install torch1.8.0+cu111 torchvision0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
重新安装torch1.8.0和cuda11.1

你可能感兴趣的:(目标跟踪,tensorflow,pytorch,神经网络)