首页 > python教程

从零开始的TensorFlow+VScode开发环境搭建的步骤(图文)

时间:2020-10-10 python教程 查看: 1091

VScode不愧是用户数量上升最快的编辑器,界面华丽(当然,需要配合各种主题插件),十分容易上手且功能强大。之前用它写C++体验十分nice,现在需要学习tensorflow,而工欲善其事必先利其器,搭建一个舒服的开发环境是非常重要的第一步。

目标:在linux下从无到有,安装anaconde3, tensorflow, 配置vs code,直到可以运行一个tensorflow版hello world(实为mnist手写数据分类)。尽管有其他的安装tensorflow的方式,但使用anaconda安装最方便,而且在conda虚拟环境中安装不怕搞砸重来,反正是虚拟环境嘛。

安装Anaconda3

下载Anaconda

对国内用户来说,最快最方便的方式就是到清华镜像站上下载anaconda的安装包:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

安装Anaconda

打开终端,cd到下载目录。下载得到的文件具有.sh后缀,显然是一个shell脚本,于是用bash命令执行它:

bash Anaconda3-5.3.1-Linux-x86_64.sh 

以下除非特殊说明,命令均为在bash下执行。

注意这里的文件名可能与你下载的文件名字不同,务必换成自己下载的文件名。

安装过程中会有一些需要确认的问题,一路yes即可。

配置环境变量

安装程序会提示你将Anaconda加入$PATH,如果你没有加入,或忘记了自己是否选择了YES,那么需要配置环境变量。

用任意的编辑器打开.bashrc文件(需要sudo 权限才能编辑),在文件末尾添加一行

export PATH="/home/lanhsi/anaconda3/bin:$PATH"
#注意,lanhsi是我的用户名,如果你也将Anaconda安装在默认位置,那么将lanhsi替换成你自己的用户名即可,否则需要将地址整个替换为anaconda的安装位置。

退出编辑器,执行

source ~/.bashrc

这条命令将重新载入你的bash配置,以使环境变量修改生效。

检验安装是否成功:

conda -V

如果像下图一样打印出conda版本号,说明安装成功。

安装VS CODE

实际上安装Anaconda时,安装程序会声称anaconda与微软具有合作关系,然后询问你是否要安装VS CODE,如果你在这一步没有选择安装VS CODE,那么就需要自行安装了。

由于linux各发行版的安装方式都不同,因此这里不详细描述安装过程了。对于我使用的ubuntu19.10来说,从vs code官网下载.deb 包,然后终端运行

dpkg -i 包名.deb

即可

安装TENSORFLOW

首先,创建一个叫做tf的虚拟环境

conda create -n tf

然后,我们需要进入这个刚创建的tf环境中去安装tensorflow

source activate tf
#注:如要退出这个环境,只需要执行source deactivate tf

进入虚拟环境后,提示符会变化,显示你正处在的虚拟环境名字

让conda去安装tensorflow,以及tensorflow依赖的一大堆其他软件

conda install tensorflow

过程比较漫长,可以先喝杯咖啡休息一下。

安装成功后,进入tf虚拟环境,进入python命令行,尝试导入tensorflow:

import tensorflow as tf

没有报错,那么说明tensorflow安装好了。

配置VSCODE

距离万事大吉只差一步了,只要在VS CODE中添加CONDA的环境,就能在VS CODE中方便的使用tensorflow了。

假定我们已经安装好了官方Python插件。打开VS CODE,点击文件->首选项,点击左侧列表中python旁边的小齿轮,选择配置扩展信息

找到右边的配置选项autoComplete:extraPath,选择在settings.json中编辑

图中选中的三行是我们需要添加的信息。这几行中,第一行自定义了一个Python解释器,将其设置为anaconda带的python解释器;后面几行配置了Python自动补全的路径,相当于是include了这些目录中的文件,这样VS CODE就可以读取到tensorflow,方便自动补全。同样,记得将路径中的lanhsi换成你的用户名。

不要忘了在配置完成后,更换Python解释器为图中下拉清单的最后一个也就是在settings.json中设置的那个,相当于将运行环境更换为我们刚添加的conda虚拟环境。

创建一个py文件,内容为

import tensorflow as tf
mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation='softmax')
])


model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)


model.evaluate(x_test,  y_test, verbose=2)

最后一步,在VSCODE的集成终端中运行这个文件。

如果配置成功,新建终端时,可以看到自动启动了conda 且进入了tf虚拟环境:

code runner插件更是可以一步直接运行文件,非常方便。

到此VS CODE下的TENSORFLOW环境就搭好了。万事开头难,而一个好的开头又是成功的一半,环境搭好后我们的项目就等于已经开发了一半了(笑)

到此这篇关于从零开始的TensorFlow+VScode开发环境搭建的步骤(图文)的文章就介绍到这了,更多相关TensorFlow VScode 环境搭建内容请搜索python博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持python博客!

展开全文
上一篇:python自动提取文本中的时间(包含中文日期)
下一篇:Python函数__new__及__init__作用及区别解析
输入字:
相关知识
Python 实现图片色彩转换案例

我们在看动漫、影视作品中,当人物在回忆过程中,体现出来的画面一般都是黑白或者褐色的。本文将提供将图片色彩转为黑白或者褐色风格的案例详解,感兴趣的小伙伴可以了解一下。

python初学定义函数

这篇文章主要为大家介绍了python的定义函数,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助,希望能够给你带来帮助

图文详解Python如何导入自己编写的py文件

有时候自己写了一个py文件,想要把它导入到另一个py文件里面,所以下面这篇文章主要给大家介绍了关于Python如何导入自己编写的py文件的相关资料,需要的朋友可以参考下

python二分法查找实例代码

二分算法是一种效率比较高的查找算法,其输入的是一个有序的元素列表,如果查找元素包含在列表中,二分查找返回其位置,否则返回NONE,下面这篇文章主要给大家介绍了关于python二分法查找的相关资料,需要的朋友可以参考下