怎么测试tensorflow安装是否成功

怎么测试tensorflow安装是否成功,第1张

首先使用命令:

pip show tenwsorflow

查看tensorflow的相关信息是否存在。

然后进入python命令行模式(在终端直接敲python回车),然后输入:

import tensorflow

查看结果即可,没有显示任何信息,则表明tensorflow已经装好,如果显示如下信息,则表示tensorflow安装失败。

>>> import tensorflow

Traceback (most recent call last):

  File "<stdin>", line 1, in <module>

ModuleNotFoundError: No module named 'tensorflow'

用常规命令:pip install tensorflow -i http://mirrors.aliyun.com/pypi/simple/

安装TensorFlow后,会出现如下提示:

即官方Build版,没有开启Intel CPU优化,估计是为了兼容AMD的CPU吧

若使用Intel CPU且确认支持AVX AVX2,可以安装Intel优化版的TensorFlow;训练和推理速度可以得到极大提升。安装Intel优化版TensorFlow的步骤:

安装完毕后,运行测试程序

TensorFlow V1.x版本,检测是否使用Intel MKLDNN优化,用命令:

TensorFlow V2.x版本,检测是否使用Intel MKLDNN优化,用命令:

新版tensorflow-gpu, 2.2.0 版推荐的测试程序与之前的1.2.0 不同,记录一下。

(1) jupyter notebook示例

(2)pycharm示例

(1) jupyter notebook

(2) pycharm 测试示例


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/yw/11656616.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-05-17
下一篇 2023-05-17

发表评论

登录后才能评论

评论列表(0条)

保存