问题:
使用YOLOv5 训练模型时,报错
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
原因:设置的批大小batch-size过大
解决方法:根据自己显卡的显存(通过nvidia-smi查看),降低batch-size,惯例是设置成2的幂次方(当然,不是2的幂其实也无妨)
问题:
使用YOLOv5 训练模型时,报错
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
原因:设置的批大小batch-size过大
解决方法:根据自己显卡的显存(通过nvidia-smi查看),降低batch-size,惯例是设置成2的幂次方(当然,不是2的幂其实也无妨)