tensorflow安装-【老鱼学tensorflow2】

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')))
posted @ 2020-07-16 15:46  dreampursuer  阅读(163)  评论(0编辑  收藏  举报