【问题标题】:Do I need to install CUDA driver for tensorflow-gpu manually if I install tf through conda如果我通过 conda 安装 tf,是否需要手动为 tensorflow-gpu 安装 CUDA 驱动程序
【发布时间】:2019-04-16 21:50:20
【问题描述】:

我按照本教程使用 conda (https://www.pugetsystems.com/labs/hpc/The-Best-Way-to-Install-TensorFlow-with-GPU-Support-on-Windows-10-Without-Installing-CUDA-1187/) 安装了 tf-gpu,它工作正常,因为我在打印的日志中看到“...gpu:0”。在我安装之前,我已经安装了CUDA驱动,所以我不确定。

在我看来 conda install tensorflow-gpu 带有 cuda 工具包和 cuDNN 等。我想知道安装 CUDA 驱动程序是否是必需的步骤。我发现的另一篇文章也没有提到驱动程序 (https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc)。但是官方GPU指南说这是必需的,所以我很困惑。我在 Windows 10 上做。

【问题讨论】:

  • 根据我在 linux 上的经验,conda install cuda 不会安装 GPU 驱动程序。并且需要 GPU 驱动程序才能使用 CUDA GPU。在 Windows 上,在许多情况下,Windows 会自动安装 GPU 驱动程序,例如适用于典型的 WDDM GPU,如 GeForce GPU。这个驱动与conda安装的工具包版本是否兼容可能无法保证,但我认为一般是兼容的。

标签: python tensorflow cuda deep-learning


【解决方案1】:

根据我的经验,您不需要安装 cuda 或 cudnn。只要你的图形驱动程序就足够了。

但根据您的系统,它可能没有得到优化。为此,您需要从头开始编译 tensorflow 并针对您的系统进行优化。

【讨论】:

    【解决方案2】:

    取决于您正在运行的机器。例如,您可以configure a Google Deep Learning VM to install the NVIDIA driver on startup

    如果未安装驱动程序,请按照有关如何安装 NVIDIA 驱动程序的 Tensorflow 说明进行操作。 Here are the instructions for Linux。请注意,您只需要安装驱动程序,而不需要安装工具包。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-07-02
      • 1970-01-01
      • 1970-01-01
      • 2015-09-10
      • 1970-01-01
      相关资源
      最近更新 更多