为什么 conda 无法在 Windows 上正确安装 tensorflow gpu？

问题描述

我在工作时使用 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 那样处理它？

解决方法

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

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