【发布时间】：2021-07-06 18:44:40
【问题描述】：
我正在尝试将 GPU 与 Tensorflow 一起使用。我的 Tensorflow 版本是2.4.1，我使用的是 Cuda 版本 11.2。这是nvidia-smi 的输出。
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.39 Driver Version: 460.39 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce MX110 Off | 00000000:01:00.0 Off | N/A |
| N/A 52C P0 N/A / N/A | 254MiB / 2004MiB | 8% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1151 G /usr/lib/xorg/Xorg 37MiB |
| 0 N/A N/A 1654 G /usr/lib/xorg/Xorg 136MiB |
| 0 N/A N/A 1830 G /usr/bin/gnome-shell 68MiB |
| 0 N/A N/A 5443 G /usr/lib/firefox/firefox 0MiB |
| 0 N/A N/A 5659 G /usr/lib/firefox/firefox 0MiB |
+-----------------------------------------------------------------------------+
我遇到了一个奇怪的问题。以前，当我尝试使用tf.config.list_physical_devices() 列出所有物理设备时，它正在识别一个 cpu 和一个 gpu。之后我尝试在 GPU 上进行简单的矩阵乘法。它因这个错误而失败：failed to synchronize cuda stream CUDA_LAUNCH_ERROR（错误代码是这样的，我忘了记下它）。但在那之后，当我再次从另一个终端尝试同样的事情时，它无法识别任何 GPU。这一次，列出物理设备会产生这样的结果：
>>> tf.config.list_physical_devices()
2021-04-11 18:56:47.504776: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2021-04-11 18:56:47.507646: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2021-04-11 18:56:47.534189: E tensorflow/stream_executor/cuda/cuda_driver.cc:328] failed call to cuInit: CUDA_ERROR_UNKNOWN: unknown error
2021-04-11 18:56:47.534233: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: debadri-HP-Laptop-15g-dr0xxx
2021-04-11 18:56:47.534244: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: debadri-HP-Laptop-15g-dr0xxx
2021-04-11 18:56:47.534356: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:200] libcuda reported version is: 460.39.0
2021-04-11 18:56:47.534393: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:204] kernel reported version is: 460.39.0
2021-04-11 18:56:47.534404: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:310] kernel version seems to match DSO: 460.39.0
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
我的操作系统是 Ubuntu 20.04、Python 版本 3.8.5 和 Tensorflow ，如之前提到的 2.4.1 和 Cuda 版本 11.2。我从these 指令安装了cuda。一项附加信息；当我导入 tensorflow 时，它显示以下输出：
import tensorflow as tf
2021-04-11 18:56:07.716683: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
我错过了什么？为什么 GPU 之前可以识别，却无法识别？
【问题讨论】：
-
这些是必需的版本。 tensorflow.org/install/source#gpu
-
@papaya 我的配置不正确吗？我想我正在使用链接中提到的版本
-
安装 CUDA toolkit 11.0 并在
sudo apt-get install nvidia-modprobe之后重新启动。谢谢 -
我有 tensorflow 2.5 和 cuda 11.0 并得到相同的错误“调用 cuInit 失败：CUDA_ERROR_UNKNOWN：未知错误”我错过了什么
-
@Ricky 你的 nvidia-smi 显示 CUDA 版本：11.2 并且 import tensorflow 显示 libcudart.so.11.0。为什么这些版本不同？根据tensorflow.org/install/source#gpu，TensorFlow和CUDA的版本应该兼容
标签： python-3.x gpu tensorflow2.0 nvidia