第一篇文章从tensorflow2.0的安装开始

231 阅读1分钟

本人也是根据同学的引导从某站转战过来,最近准备全身心投入深度学习的研究中,因此在新的平台中从0开始,继续奋斗。

对于python的安装此处不做过多的介绍,tensorflow2.0推出已经有很长一段时间了,相比于1.0的版本,2.0更加的模块化,语言也简练了不少。由于目前本人使用的是windows10系统,所以本文的讲解均已windows10为例。

1.安装tensorflow2.0

在dos窗口中输入下面的代码安装tensorflow2.0

CPU版 pip install tensorflow==2.7.0 -i https://pypi.doubanio.com/simple

GPU版 pip install tensorflow-gpu==2.7.0 -i https://pypi.doubanio.com/simple

安装GPU版本时一定要确保电脑中安装了CUDA,不然安装后可能仍然无法正常使用。本人的电脑中安装了CUDA10,因此我选则了安装GPU版本,安装过程如下

安装.PNG

安装完成后接下来就是验证是否安装成功,只需要在IDE(Pycharm等)中新建一个测试文件,输入以下代码测试即可。

import tensorflow as tf

# 权重
w = tf.Variable(tf.ones(shape=(3, 3)), name="W")
# 偏置
b = tf.Variable(tf.zeros(shape=(3)), name="b")

@tf.function
def model(x):
    return w * x + b

rest = model([1, 2, 3])
print(rest)

如果输出下面的结果,则表示安装成功。

1.PNG

上面是一个简单的矩阵运算,可以看到tensorflow对于运算方面代码非常简洁,主要原因是后端进行了大量的封装,使我们使用起来更加方便,相对于1.0版本,代码量缩短了接近一倍。