E tensorflow/stream_executor/cuda/cuda_driver.cc:351对cuInit的调用失败: CUDA_ERROR_NO_DEVICE:没有检测到具有CUDA功能的设备

Question

我已经安装了GTX 1660超级卡。然后用cudnn 7.6.4.38安装CUDA10.1，并运行以下代码‘import 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，shape=2，3，name='a') b= tf.constant(1.0，2.0，3.0，4.0，5.0，6.0，shape=3，2，name='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:没有检测到具有数据自动化功能的设备

我试图按以下方式更改代码，但没有工作

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驱动程序是最新的，或者相应地更新它。