在 GPU 上运行 Keras 模型

Question

我在 Windows 10 上运行并已安装 CUDA 12.3

喀拉斯 = 2.15.0 张量流 = 2.15.0

我的模型只是在 CPU 上训练，而不是在 GPU 上训练。还有代码：

from tensor flow.python.client import device_lib
print(device_lib.list_local_devices())

我得到：

WARNING: tensorflow: From C: \envs\p311_deeplearning\Lib\site-packages\keras\src\losses.py: 2976: The name tf.losses.sparse_softmax_cross_entropy is deprecate
d. Please use tf.compat.v1. losses sparse_softmax_cross_entropy instead.

[name: "/device: CPU: 0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 14870411986114771269
xIa_global_id: -1

如何才能找到我的 GPU 并在其上进行训练？ （GPU = NVIDIA RTX 2080 Ti）

Answer 1

我找到了解决方案。

安装Conda只是第一步。之后，必须将 CUDNN 放置在硬盘驱动器上，并且必须设置环境变量。我错过了后者。

以下视频描述了更好的处理方式，比我写下来的要好：

https://www.youtube.com/watch?v=OEFKlRSd8Ic