Computer vision pytorch-yolov3列车运行时错误:无法找到有效的cuDNN算法来运行卷积
当我开始使用“python train.py”来训练yolov3时,发生了这种情况,原因是什么Computer vision pytorch-yolov3列车运行时错误:无法找到有效的cuDNN算法来运行卷积,computer-vision,pytorch,object-detection,yolo,cudnn,Computer Vision,Pytorch,Object Detection,Yolo,Cudnn,当我开始使用“python train.py”来训练yolov3时,发生了这种情况,原因是什么 Traceback (most recent call last): File "train.py", line 415, in <module> train() # train normally File "train.py", line 279, in train scaled_loss.back
Traceback (most recent call last):
File "train.py", line 415, in <module>
train() # train normally
File "train.py", line 279, in train
scaled_loss.backward()
File "C:\Users\zst\anaconda3\lib\site-packages\torch\tensor.py", line 198, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "C:\Users\zst\anaconda3\lib\site-packages\torch\autograd\__init__.py", line 100, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution (try_all at ..\aten\src\ATen\native\cudnn\Conv.cpp:693)
(no backtrace available)
回溯(最近一次呼叫最后一次):
文件“train.py”,第415行,在
训练()#正常训练
文件“train.py”,第279行,列车中
缩放损失。向后()
文件“C:\Users\zst\anaconda3\lib\site packages\torch\tensor.py”,第198行,向后
火炬。自动标记。向后(自我、梯度、保留图、创建图)
文件“C:\Users\zst\anaconda3\lib\site packages\torch\autograd\\uuuuu init\uuuuu.py”,第100行,向后
allow_unreachable=True)#allow_unreachable标志
RuntimeError:找不到有效的cuDNN算法来运行卷积(请在..\aten\src\aten\native\cuDNN\Conv.cpp:693处重试)
(无回溯可用)
win10 2080ti cuda10.2 cudnn7604 python 3.7 torch1.5 torchvision0.6
跟踪屏幕截图:环境截图:
您能想出一个更简单的测试用例吗?如果您的cudnn断开,即使是一个带有随机输入的简单1 conv层网络也会断开。你能把问题缩小到一个小的可重复的例子吗?你能想出一个更简单的测试用例吗?如果您的cudnn断开,即使是一个带有随机输入的简单1 conv层网络也会断开。你能把问题缩小到一个可重复的小例子吗?