首页 > 解决方案 > 如何使用 tensorflow-gpu?

问题描述

我正在尝试将我的 GPU 用于 keras-tensorflow。但是我找不到GPU。

import tensorflow as tf
print("Num GPU Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

输出:Num GPU 可用:0

当我试图找到CPU时,

import tensorflow as tf
print("Num CPU Available: ", len(tf.config.experimental.list_physical_devices('CPU')))

输出:Num CPU 可用:1

有人可以解释有什么区别吗?如何使用我的 GPU?我有 2 个“GT 750M”GPU。

我安装了 CUDA v10.1 和 cuDNN-10.1-windows7-x64-v7.6.5.32.zip。我也在使用 tensorflow 1.14.0 和 numpy 1.16.4

标签: tensorflowimage-processingnvidia

解决方案


推荐阅读