首页 > 解决方案 > 安装 Tensorflow 2.0,我有 GPU 但没有启用 CUDA,这是否意味着我只能使用我的 CPU?

问题描述

我阅读了有关如何在 Windows 上使用 GPU 安装 Tensorflow 2.0 的 TensorFlow 文档(https://www.tensorflow.org/install/gpu)。

在硬件要求部分,它声明它需要具有 CUDA 计算能力的 NVIDIA GPU。当我看到支持 CUDA 的 GPU 卡列表(https://developer.nvidia.com/cuda-gpus)时,我才意识到我的 GPU 不在列表中(我的机器上有 NVIDIA GTX 1660 ti)。

我想问的是:

  1. 这是否意味着我不能将我的 GPU 用于我的深度学习 Tensorflow 实施?
  2. 是不是意味着我必须安装Tensorflow 2.0的CPU版本?

标签: pythontensorflow

解决方案


您可以在笔记本电脑上使用 Tensorflow GPU,因为您的 GPU 支持 CUDA。安装 CUDA,然后是 cuDNN,然后是 Tensorflow GPU。

您不必使用 CPU 版本,因为您有更好的选择。


推荐阅读