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

  • 5
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值