search

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

您好， 我想问下您的cuda是什么版本的啊，我这个一直报错 #86

Open lukaisheng1203 opened 5 years ago

Stinky-Tofu commented 5 years ago

10.0

lukaisheng1203 commented 5 years ago

谢谢！我也是10.0.130，cudnn 7.4，但是却一直报错，提示 Could not create cudnn handle:CUDNN_STATUS_INTERNAL_ERROR UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. 不知道您有没有遇到过这样的情况啊，我处理了半天都美解决。。。

Stinky-Tofu commented 5 years ago

不知道啊，这个项目好久没跑过了

lukaisheng1203 commented 5 years ago

谢谢您！我限制了显存使用，发现是显存不够，我的显卡是2070， 8G，可是我的batch size只是设置为1啊，这么好的工程，我却跑不了自己的数据集，好难受