E tensorflow/stream_executor/cuda/cuda_driver.cc:351 调用 cuInit 失败：CUDA_ERROR_NO_DEVICE：未检测到支持 CUDA 的设备

Question

我已经安装了GTX 1660 Super Card。然后用cudnn 7.6.4.38安装CUDA 10.1 当我运行下面的代码`import tensorflow as tf 导入 os os.environ["CUDA_DEVICE_ORDER"] = "PCI bus 1" os.environ["CUDA_VISIBLE_DEVICES"] = "1"

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], 形状=[2, 3], 名称='a') b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], 形状=[3, 2], 名称='b') c = tf.matmul(a, b)

打印(c)`

我收到以下错误 2020-04-28 19:59:34.933933：E tensorflow/stream_executor/cuda/cuda_driver.cc:351] 调用 cuInit 失败：CUDA_ERROR_NO_DEVICE：未检测到支持 CUDA 的设备

我尝试如下更改代码但没有成功

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

也试过

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

Answer 1

查找CUDA、Tensorflow、Python、cuDNN的兼容版本，可以访问官网：www.tensorflow.org/install/source#gpu.
请检查您的版本是否正确。

（题外话）如果您也在使用 Docker 容器：我会推荐这样的图片：nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04

其中：

CUDA_VERSION=10.1.243
CUDNN_VERSION=7.6.5.32

还要确保你有兼容的 tf 版本，python:

Tensorflow-2.2.0 或 2.3.0
Python 3.5-3.8

接下来要确保您的 GPU 驱动程序是最新的，或者进行相应的更新。

E tensorflow/stream_executor/cuda/cuda_driver.cc:351 调用 cuInit 失败：CUDA_ERROR_NO_DEVICE：未检测到支持 CUDA 的设备

E tensorflow/stream_executor/cuda/cuda_driver.cc:351 failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected

python-3.x

tensorflow2.0