在gpu上运行pyorch代码时出现cudnn错误
在gpu上运行pyorch代码时出现cudnn错误
提问于 2019-03-01 00:31:55
我有以下错误: Traceback (最近一次调用)：
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这不是我的代码。我只是在尝试运行最近的一篇论文“神经常微分方程”-by Chen等人的代码。这是指向code的链接。
gpu: nvidia 2080Ti
pytorch版本：‘1.0.1.post2’
cuda 9.0
python 3.7.2
cudnn
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 4
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)我是pytorch的新手。为什么我会得到这个错误，我如何修复它？
回答 1
Stack Overflow用户
回答已采纳
发布于 2019-03-01 04:05:29
RTX2080Ti需要CUDA10来处理包含CUDA10的PyTorch二进制文件
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/54930268
复制相关文章
相似问题