python - 如果我通过conda安装tf,是否需要手动安装tensorflow-gpu的CUDA驱动

标签 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 上执行此操作。

最佳答案

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

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

关于python - 如果我通过conda安装tf,是否需要手动安装tensorflow-gpu的CUDA驱动,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53291571/

相关文章:

c++ - std::isnan 尚未声明,带有 g++-5

python - 如何在不同目录下运行python?

c++ - CUDA 使用 cudaMemcpy 复制多个结构数组

python - 加速 nautilus python-extensions 以读取图像的 Exif

python - 重复推理后模型推理运行时间增加

python - ValueError : Error when checking input: expected dense_1_input to have shape (24, )但得到形状为(1,)的数组

tensorflow - 在 Tensorflow 中添加整个训练/测试数据集的准确性摘要

clojure - Clojure上的GPU编程?

php - 允许 python 脚本创建文件的最佳/最简单的 Apache 服务器/文件夹设置是什么?

python - pip 如何知道 virtualenv 以及在哪里安装包