TensorFlow极简安装

本文介绍TensorFlow 2.0的安装,TensorFlow 2.0兼容64位Python版本

  • Python 3.5—3.7
  • Python 2.7(windows平台不支持)

本文使用3.7
        TensorFlow 2.0分为CPU和GPU两个版本,其中GPU版本的TensorFlow可以利用NVIDIA GPU强大的计算加速能力,使Tensorflow的运行更为高效,尤其是可以成倍提升模型训练的速度。

1、tensorflow-CPU安装

命令语句:pip install tensorflow
注:pip版本不低于19.0
pip是一个现成的,通用的Python包管理工具,提供了对Python包的查找、下载、安装、卸载的功能。

2、tensorflow-GPU安装

    tensorflow对NVIDIA显卡的支持较为完备。AMD的显卡也开始对tensorflow提供支持。
注:(1)NVIDIA显卡要求其CUDA Compute Capability不低于3.0
          算力参考博文
          算力参考:官网
(2)NVIDIA驱动程序需410.x或更高版本
          命令语句:nvidia-smi
          如果遇到如下错误:‘nvidia-smi’ 不是内部或外部命令,也不是可运行的程序
          则需要更改命令路径:cd C:\Program Files\NVIDIA Corporation\NVSMI

安装tensorflow 2.0的命令行语句:conda install tensorflow-gpu==2.0.0

最后在Jupyter NoteBook中检查版本:
import tensorflow as tf
tf.test.is_gpu_available()

返回为:true

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值