我在工作中使用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 one那样处理它?
发布于 2021-05-05 01:02:03
我对conda的工作原理没有太多的专业知识,因为我已经手动下载并使用CUDA和cuDNN配置了tensorflow。
为了回答你的问题,我建议你浏览这个博客,Setting up Tensorflow-GPU with Cuda and Anaconda on Windows。在这里,用户使用的命令与您提到的命令不同。还涉及到一些额外的步骤。
https://stackoverflow.com/questions/67388244
复制相似问题