如何在 Windows 10 上运行 tensorflow？ 我有 GPU Geforce gtx 1650。我可以在上面运行 tensorflow 吗？ 如果是，那么如何？

Question

我想在我的电脑上用 Python 做一些机器学习，我在安装 tensorflow 时遇到了问题，我发现 tensorflow 可以与启用 CUDA 的 GPU 一起使用。 我有一个 GPU Geforce gtx 1650，tensorflow 会处理这个问题。

如果是，那么，我怎么能这样做呢？

Answer 1

以管理员模式打开命令提示符后，支持GPU的Tensorflow的安装命令如下：

    pip3 install --upgrade tensorflow-gpu 

要检查 tensorflow 是否已成功安装，请使用命令：

import tensorflow as tf

要测试 CUDA 对 Tensorflow 安装的支持，您可以在 shell 中运行以下命令：

    tf.test.is_built_with_cuda()  

[警告：如果安装了非 GPU 版本的软件包，该函数也将返回 False。 使用此命令来验证 TensorFlow 是否是使用 CUDA 支持构建的。]

最后，为了确认 GPU 可用于 Tensorflow，您可以使用 TensorFlow 中的内置实用程序函数进行测试，如下所示：

    tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)

Answer 2

安装 tensorflow-gpu 在 GPU 上进行计算。 您可以使用下面的代码来检查您的 GPU 是否正在被 tensorflow 使用。

tf.test.is_gpu_available(
cuda_only=False,
min_cuda_compute_capability=None

)

Answer 3

下面是安装tensorflow的步骤：

1. 下载并安装 Visual Studio。
2. 安装 CUDA 10.1
3. 将 lib、include 和 extras/lib64 目录添加到 PATH 变量。
4. 安装cuDNN
5. 通过pip install tensorflow

Answer 4

我不认为你可以。 https://www.tensorflow.org/install/gpu Tensorflow 明确提到了支持的架构列表，遗憾的是 1650 不属于该列表。 检查上面网站上的“启用 cuda 的 gpu 卡”链接。