tensorflow版本已经升级到了2.x,其在易用性上大大增强,主要是由于API采用了keras的方式,大大简化了开发工作。
安装
pip3 install tensorflow
验证
输入如下代码:
import tensorflow as tf
如果没有出错信息显示就表示安装成功了!
是否很简单:)
我们还可以通过Anaconda来安装tensorflow。
首先,下载Anaconda,安装完成之后,运行anaconda navigator图形界面,点击其中的Environments创建一个新的环境,比如命名成tensorflow
,其中会弹出对话框需要设置python版本,目前选择为3.7,点击确定后就创建了一个新的python环境。
接着就可以选择在All中搜索tensorflow来进行安装,如果有GPU,则还可以搜索tensorflow_gpu版本的,这样就可以支持gpu来运行(前提假设你有NVIDIA GPU)。
在运行时如果报告显卡驱动版本不够高的错误的话,可以直接在如下的网址中点击NVIDIA® GPU 驱动程序来下载最新的驱动程序。
https://tensorflow.google.cn/install/gpu
可以通过如下的语句来打印GPU的个数:
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))