如果通过conda安装TF,我是否需要手动为tensorflow-gpu安装CUDA驱动程序

时间:2018-11-14 00:43:59

标签: python tensorflow cuda deep-learning

我遵循了本教程,并使用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上执行此操作。

2 个答案:

答案 0 :(得分:1)

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

但是根据您的系统,可能未进行优化。为此,您需要从头开始编译tensorflow并针对您的系统对其进行优化。

答案 1 :(得分:0)

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

如果未安装驱动程序,请按照Tensorflow指导说明如何安装NVIDIA驱动程序。 Here are the instructions for Linux。请注意,您只需要安装驱动程序,而无需安装工具箱。