【发布时间】: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 那样处理它?
【问题讨论】:
标签: windows tensorflow conda nvidia