GPU WIndows 11 上的 Tensorflow

Question

我有 Windows 11、nVidia GTX 1650 Ti，我正在尝试让 Tensorflow 在 GPU 上工作。还有一个绝对条件：我什至不想考虑 Linux。它必须在 Windows 上运行。目前使用 Python 3.11.2、Tensorflow 2.13.1

我尝试了 nVidia 官方指南来安装 cuda 和 cudnn，以及medium.com 上的一些文章，但仍然一无所获。我从 PyCharm 切换到 Visual Studio Code。没有什么。通过 conda 安装。没有什么。我尝试了最新的 cuda 和 cudnn 版本，cuda 11.8 和 cudnn 8.6.0 或 cudnn 8.9.7，但仍然没有任何结果。 Tensorflow 就是看不到我的 GPU。是的，nVidia 游戏驱动程序已更新。除了linux我没有其他的想法，但我宁愿从窗口跳下去也不愿使用linux。我是一名数学家，不是 IT 专家，所以请给我逐步的解决方案。预先感谢

import sys
import keras
import tensorflow as tf
import numpy as np

print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {keras.__version__}")
print()
print(f"Python {sys.version}")
gpu = len(tf.config.list_physical_devices('GPU'))>0
print("GPU is", "available" if gpu else "NOT AVAILABLE")

Answer 1

您需要将 TF 版本降级到 2.10 或更低版本才能在本机 Windows 上工作（但不推荐）

本机 Windows 上的 GPU 支持仅适用于 2.10 或更早版本，从 TF 2.11 开始，Windows 不支持 CUDA 构建。 Tensorflow 2.10 正确支持的 CUDA 和 cuDNN 版本是 CUDA 11.2 和 cuDNN 8.1，如本测试构建配置中所述

链接：https://discuss.tensorflow.org/t/issues-installing-tensorflow-on-windows-11-with-nvidia-2060/19663

注意：TensorFlow 2.10 是在本机 Windows 上支持 GPU 的最后一个 TensorFlow 版本。从 TensorFlow 2.11 开始，您需要在 WSL2 中安装 TensorFlow，或者安装 TensorFlow 或 TensorFlow-CPU，并且可以选择尝试 TensorFlow-DirectML-Plugin

链接：https://www.tensorflow.org/install/pip#windows-native