- ubantu16.04解决报错:Check failed: stream->parent()->GetConvolveAlgorithms 已放弃 (核心已转储)
- 报错如下:
E tensorflow/stream_executor/cuda/cuda_dnn.cc:378] Loaded runtime CuDNN library: 7402 (compatibility version 7400) but source was compiled with 7004 (compatibility version 7000).
If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
2019-05-01 19:20:26.894925: F tensorflow/core/kernels/conv_ops.cc:717] Check failed:
stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo(), &algorithms)
已放弃 (核心已转储)
- 解决办法:
- 在python的执行文件中,加入以下代码即可:
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="1"