Pytorch 在gpu上运行pyorch代码时出现cudnn错误
我有以下错误: 回溯(最近一次呼叫最后一次):Pytorch 在gpu上运行pyorch代码时出现cudnn错误,pytorch,Pytorch,我有以下错误: 回溯(最近一次呼叫最后一次): 我是个新手。为什么会出现此错误以及如何修复它?RTX2080Ti需要CUDA10才能正常工作。安装包含CUDA10的PyTorch二进制文件如何计算GPU所需的最低cuda版本? File "odenet_mnist.py", line 343, in <module> logits = model(x) File "/home/subhashnerella/.conda/envs/pytorch/lib/python3
我是个新手。为什么会出现此错误以及如何修复它?RTX2080Ti需要CUDA10才能正常工作。安装包含CUDA10的PyTorch二进制文件如何计算GPU所需的最低cuda版本?
File "odenet_mnist.py", line 343, in <module>
logits = model(x)
File "/home/subhashnerella/.conda/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/home/subhashnerella/.conda/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/home/subhashnerella/.conda/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/home/subhashnerella/.conda/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 320, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 4
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)