Tensorflow 未使用/查看我的 GPU！ （在窗户上）

Question

更新我的张量流版本后，我无法使用 GPU 进行 NN 训练等。 Tensorflow 没有看到也没有使用我的 GPU，即使 CUDA 驱动程序已更新、正确安装，并且我可以使用 Nvidia 的工具 nvidia-smi.exe 看到我的 GPU。我的系统运行 Window 10 和 Anaconda。

Answer 1

在 Tensorflow 2.11 版及更高版本中，Windows 上没有（本机）GPU 支持。这意味着如果您希望 Tensorflow 查看并使用 GPU，则需要安装 Tensorflow 版本 2.10。根据文档，TF v2.10+自带GPU功能，但实际情况是很多情况下看不到GPU，需要显式安装gpu-Tensorflow。 Tensorflow v.2.10 与 Python v.3.9-3.10 兼容。

您应该使用Python v.3.9-3.10通过Anaconda创建一个新环境并安装tensorflow v.2.10。然后在该环境中打开 anaconda 命令提示符并运行

pip install tensorflow-gpu==2.10.0

。要确认您的 GPU 现在对 TF 可见/可用，请使用以下代码创建一个 Python 脚本并运行它。

import tensorflow as tf
print(f'TF version: {tf.__version__}')

print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
tf.debugging.set_log_device_placement(True)

gpus = tf.config.experimental.list_physical_devices('GPU')

if gpus:
    # Extract the GPU indices from the device names
    gpu_indices = [int(gpu.name.split(':')[-1]) for gpu in gpus]
    print("GPU Indices:", gpu_indices)
else:
    print("No GPUs available in your system.")

您现在应该得到

Num GPUs Available = 1

。

Answer 2

非常感谢，对我的作品帮助很大