RuntimeError: cuDNN版本不匹配: PyTorch是针对7102编译的，但与7604链接在一起
RuntimeError: cuDNN版本不匹配: PyTorch是针对7102编译的，但与7604链接在一起
提问于 2020-02-10 14:11:20
我在运行深度学习模型训练时遇到了这个错误，尽管在Internet上查看了许多解决方案，但它们对我没有帮助。日志如下：
Traceback (most recent call last):
File "main.py", line 208, in <module>
main()
File "main.py", line 100, in main
model = nn.DataParallel(model).cuda()
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/parallel/data_parallel.py", line 105, in __init__
self.module.cuda(device_ids[0])
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/module.py", line 249, in cuda
return self._apply(lambda t: t.cuda(device))
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/module.py", line 176, in _apply
module._apply(fn)
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/rnn.py", line 112, in _apply
self.flatten_parameters()
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/rnn.py", line 78, in flatten_parameters
if not any_param.is_cuda or not torch.backends.cudnn.is_acceptable(any_param):
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/backends/cudnn/__init__.py", line 87, in is_acceptable
if _libcudnn() is None:
File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/backends/cudnn/__init__.py", line 58, in _libcudnn
'but linked against {}'.format(compile_version, __cudnn_version))
RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7102 but linked against 7604回答 2
Stack Overflow用户
回答已采纳
发布于 2020-02-10 14:11:20
问题是，无论日志中显示的版本是什么，7.6.4是我的cudnn版本，7.1.2是代码最初编译的cudnn版本。我需要的是降级(或升级我当前的cudnn版本)：
conda install cudnn=7.1.2它工作，如果有的话，请纠正我。
Stack Overflow用户
发布于 2020-05-01 17:24:12
我有一个类似的problem...downgrading cudnn为我工作。正如这里提到的，错误消息RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7102 but linked against 7604
...indicates Pytorch被链接到更新版本的cudnn库。这只会在我的模型中造成一个深层次的问题。我猜这是一个古老的Pytorch函数，最近没有更新。在将cudnn降级到与Pytorch相同的版本后，它again...no了其他明显的性能损失。
conda install cudnn=7.1.2
很抱歉重复，但我的StackOverflow代表太低，无法添加评论。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/60144922
复制相关文章
相似问题
腾讯云开发者
