首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如果通过conda安装tf,是否需要手动安装tensorflow-gpu的CUDA驱动程序

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

Stack Overflow用户
提问于 2018-11-14 08:43:59
回答 2查看 3K关注 0票数 1

我遵循本教程并使用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安装tensorflow-gpu附带cuda工具包和cuDNN等。我想知道安装CUDA驱动程序是否是必需的步骤。我发现的另一个帖子也没有提到driver (https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc)。但是官方的GPU指南说这是必需的,所以我很困惑。我是在Windows10上这样做的。

EN

回答 2

Stack Overflow用户

发布于 2018-11-14 16:00:22

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

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

票数 2
EN

Stack Overflow用户

发布于 2020-02-03 03:21:17

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

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

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/53291571

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档