Tensorflow（CUDA 11.2）未使用 Python 3.7 在 AMD Radeon Vega 8（Envy 笔记本电脑）上检测到 GPU

Posted 2023-04-15

【中文标题】Tensorflow（CUDA 11.2）未使用 Python 3.7 在 AMD Radeon Vega 8（Envy 笔记本电脑）上检测到 GPU

【英文标题】：Tensorflow (CUDA 11.2) not detecting GPU on a AMD Radeon Vega 8 (Envy Laptop) using Python 3.7

【发布时间】：2021-11-07 16:59:42

【问题描述】：

有没有人能够使用 python 3.7 让 tensorflow 检测 GPU？

你是怎么做到的？我已经下载了 cuDNN 8.1、CUDA 11.2，然后使用 pip install tensorflow-gpu==2.5 安装了 tensorflow 我已经为 cuDNN 的 bin 添加了另一个环境变量，但是我仍然得到这个结果 Num GPUs Available 0。 Tensorflow (CUDA 11.2) 甚至可以与 AMD Radeon Vega 8 一起使用吗？

【问题讨论】：

这主要是关于使用 AMD 硬件的 TensorFlow GPU 支持吗？

相关链接：***.com/questions/12828268/…

CUDA 是 NVIDIA 专有的 NVIDIA 技术。它不能也不能在 AMD GPU 上运行。 Tensorflow 有一个基于 rocm 的端口，可以在 AMD GPU 上运行。这不是通过正常的 Tensforlow 二进制发布渠道 AFAIK 分发的

@MohanRadhakrishnan 是的。杰么么谢谢。 talonmies 知道了，谢谢，但您知道他们可能在哪里提到它们不能在 AMD GPU 上运行吗？

【参考方案1】：

不，它不是，因为 cuDNN 是 NVIDIA 的产品，CUDA 也是。 NVIDIA 设计他们自己的 GPU，他们的产品会寻找这些 GPU。为了让 tensorflow 检测 GPU，您必须使用 NVIDIA 的 GPU。