2.0以后tensorflow和tensorflow-gpu有啥区别

pip install tensorflow 和pip install tensorflow-gpu装的东西是一样的

conda install tensorflow装的是cpu版本的tensorflow,这可能是conda清华源的问题。

想用conda 装gpu tensorflow,就要conda install tensorflow-gpu,还会自动把配套的cudatoolkit和cudnn给你装好。

当你电脑安装了cuda,或者电脑没有安装cuda只是在虚拟环境中用conda安装了cudatoolkit后,都会调用gpu跑程序,没有gpu就用cpu。

conda装东西总是喜欢检查环境中所有包的依赖关系,当你装了tensorflow再用conda装matplotlib,就把tensorflow的numpy包给替换了,造成使用tensorflow后报numpy错误,能用pip装尽量用pip。

安装gpu tensorflow最简单的方法:

conda install cudatoolkit=10.1
conda install cudnn=7.6
pip install tensorflow==2.1

你可能感兴趣的:(tensorflow)