本人也是根据同学的引导从某站转战过来,最近准备全身心投入深度学习的研究中,因此在新的平台中从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版本,安装过程如下
安装完成后接下来就是验证是否安装成功,只需要在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)
如果输出下面的结果,则表示安装成功。
上面是一个简单的矩阵运算,可以看到tensorflow对于运算方面代码非常简洁,主要原因是后端进行了大量的封装,使我们使用起来更加方便,相对于1.0版本,代码量缩短了接近一倍。