一、引言

tensorflow是常用的图像处理模型框架,结合机器学习可作用于计算机视觉。
运行环境:python 3.9,pycharm 2021.1
安装版本:tensorflow 2.6.0,tensorflow-gpu 2.6.0

二、安装

1)python 3.9版本只能适配tensorflow 2.5以上版本,在已安装好python环境下,在cmd中输入pip install tensorflow 命令可以直接安装最新适配版本,输入pip install tensorflow-gpu命令可以直接安装gpu最新版本,但需要电脑gpu达到一定配置需求才可使用。
2)在pycharm中写入import tensorflow as tf代码,若tensorflow还标红,直接在pycharm中点击install,编译器会直接进行安装。

三、测试

1)在cmd中输入命令pip show tensorflow或者pip show tensorflow-gpu可以查看当前tensorflow版本。
2)在pycharm中输入以下代码,注意在tensorflow 2版本以上的版本中,与1版本有一些语法上的区分。

import tensorflow as tf
tf.compat.v1.disable_eager_execution()
hello = tf.constant('hello,tensorflow')
sess= tf.compat.v1.Session()
print(sess.run(hello))

输出:
b’hello,tensorflow’
即为安装成功。

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐