RuntimeError: Unable to find a valid cuDNN algorithm to run convolution问题解决

问题:

使用YOLOv5 训练模型时,报错

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

原因:设置的批大小batch-size过大

解决方法:根据自己显卡的显存(通过nvidia-smi查看),降低batch-size,惯例是设置成2的幂次方(当然,不是2的幂其实也无妨)

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