pytorch升级到1.8.0后训练报错RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
pytorch升级到1.8.0后训练报错RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED@TOC
之前训练环境pytorch v1.7.1+ cuda 10.2 训练一切正常。
下载yolov5 代码后 执行了pip3 install -r requirements.txt 后,发现执行训练报错:
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
按网上的操作清空rm -rf ~.N
原创
2021-03-17 14:19:12 ·
997 阅读 ·
1 评论