首页 > 解决方案 > cudaGetDevice() 失败。状态:CUDA 驱动程序版本对于 CUDA 运行时版本不足

问题描述

当我在 GPU 中运行 tensorflow 时出现以下错误。

2018-09-15 18:56:51.011724: E tensorflow/core/common_runtime/direct_session.cc:158] Internal: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
Traceback (most recent call last):
  File "evaluate_sample.py", line 160, in <module>
    tf.app.run(main)
  File "/anaconda3/envs/tf/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "evaluate_sample.py", line 123, in main
    with tf.Session() as sess:
  File "/anaconda3/envs/tf/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1494, in __init__
    super(Session, self).__init__(target, graph, config=config)
  File "/anaconda3/envs/tf/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 626, in __init__
    self._session = tf_session.TF_NewSession(self._graph._c_graph, opts)
tensorflow.python.framework.errors_impl.InternalError: Failed to create session.

以下错误来自哪里?

E tensorflow/core/common_runtime/direct_session.cc:158] Internal: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

tensorflow.python.framework.errors_impl.InternalError: Failed to create session

这样我的版本:

张量流是:1.10

cat /proc/driver/nvidia/version

NVRM 版本:NVIDIA UNIX x86_64 内核模块 390.77 Tue Jul 10 18:28:52 PDT 2018

GCC 版本:gcc 版本 7.3.0 (Debian 7.3.0-28)

nvcc --version

nvcc:NVIDIA (R) Cuda 编译器驱动程序

版权所有 (c) 2005-2016 NVIDIA Corporation

基于 Sun_Sep__4_22:14:01_CDT_2016

Cuda编译工具,发布8.0,V8.0.44

标签: python-2.7tensorflowcuda

解决方案


更新 nvidia 驱动程序解决了这个问题。

您可以在此处检查您的 cuda 工具包兼容性。然后通过从此处下载来更新您的 nvidia 驱动程序。


推荐阅读