RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

今天跑YOLOv5-6.1时出现这个问题:
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
网络上有些说是cuda版本不对之类的,但是我比较确定我这个没有问题,因为之前一直都可以弄。后来发现是batch_size弄太大了,调小以后就可以了

你可能感兴趣的:(debug记录,深度学习,pytorch,人工智能)