为啥 conda 无法在 Windows 上正确安装 tensorflow gpu?

Posted

技术标签:

【中文标题】为啥 conda 无法在 Windows 上正确安装 tensorflow gpu?【英文标题】:Why conda cannot install tensorflow gpu properly on Windows?为什么 conda 无法在 Windows 上正确安装 tensorflow gpu? 【发布时间】:2021-07-27 00:07:32 【问题描述】:

我在工作中使用 Linux,习惯通过 Conda 安装 tensorflow GPU 版本。

Linux机器有NVIDIA驱动,nvidia-smi运行正常。

我使用的命令是

conda install python tensorflow-gpu

神奇的是,conda 可以处理所有事情。它安装 cudatoolkit 或任何相关工具。一旦我在那个虚拟环境中运行 python,GPU 就可以使用了。

现在我切换到在 Windows 上工作。 Windows 盒子有 NVIDIA 驱动程序,并且 nvidia-smi 可以正常运行。然后我使用与在 Linux 上相同的 conda 命令。 conda install 成功但虚拟环境中的python 没有检测到GPU。

How to install tensorflow 2 gpu on windows 10 with conda? 在 Windows 上说我必须手动安装 cudnn 和 Cuda。为什么 Windows 上的 conda 不像 Linux 那样处理它?

【问题讨论】:

【参考方案1】:

我对 conda 的工作原理没有太多专业知识,因为我使用 CUDA 和 cuDNN 手动下载和配置了 tensorflow。

要回答您的问题,我建议您阅读此博客Setting up Tensorflow-GPU with Cuda and Anaconda on Windows。在这里,此人使用与您提到的不同的命令。还涉及一些额外的步骤。

【讨论】:

以上是关于为啥 conda 无法在 Windows 上正确安装 tensorflow gpu?的主要内容,如果未能解决你的问题,请参考以下文章

为啥在 conda 安装后 Tensorflow 无法识别我的 GPU?

无法使用 Conda 在 Windows 上安装 Poppler

无法在 conda python 3.8 env 上正确安装 awswrangler(连接问题)

[Python[Anaconda & PyTorch]] -- 使用conda 安装 Torch 出现错误 --Windows

为啥我在 Windows 10 UWP 上收到“无法注册包”部署错误?

如何修复无法在 Conda 上加载本机 Tensorflow 运行时