tensorflow - GPU 无法识别

Question

我在带有 GPU 的 linux 18.04 上运行：

lspci | grep VGA
00:02.0 VGA compatible controller: Intel Corporation Device 3e92
01:00.0 VGA compatible controller: NVIDIA Corporation Device 1e82 (rev a1)

nvidia-smi 给出：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.119.03   Driver Version: 450.119.03   CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+

我有带有以下软件包的 python 3.6：

pip freeze |grep tensor:
tensorboard==2.5.0
tensorboard-data-server==0.6.0
tensorboard-plugin-wit==1.8.0
tensorflow==2.4.1
tensorflow-estimator==2.4.0
tensorflow-gpu==2.4.1

运行以下命令时：

len(tf.config.list_physical_devices('GPU'))

我得到 0 个 GPU

根据： https://www.tensorflow.org/install/source#gpu似乎版本是正确的。

我试着看这里：

Tensorflow 似乎没有看到我的 gpu

（重新安装 tensorflow/tensorflow-gpu）-但没有找到正确的解决方案。

我该如何解决？

标签： tensorflow