本地搭建Tensorflow环境模拟ModelArts(一)

举报
李国有 发表于 2019/07/21 22:32:52 2019/07/21
【摘要】 背景 最近参加了华为云一个非常火的活动活动“ModelArts-Lab AI实战营”目前已经进行了第四期了、此活动由浅入深慢慢让大家接触、学习、使用、最新的AI人工技术及在华为云平台中的实现方法。 第一期:图像分类 (I) https://github.com/huaweicloud/ModelArts-Lab/issues/49 第二期:图像分类 (II...

背景

    最近参加了华为云一个非常火的活动活动“ModelArts-Lab AI实战营”目前已经进行了第四期了、此活动由浅入深慢慢让大家接触、学习、使用、最新的AI人工技术及在华为云平台中的实现方法。

    

    第一期:图像分类 (I) https://github.com/huaweicloud/ModelArts-Lab/issues/49 

    

    第二期:图像分类 (II) https://github.com/huaweicloud/ModelArts-Lab/issues/113

    

    第三期:图像分类 (III) 模型参数&网络调优https://github.com/huaweicloud/ModelArts-Lab/issues/219  

    

    第四期:物体检测 (I) https://github.com/huaweicloud/ModelArts-Lab/issues/402

    活动中刚开始对平台有资源要求并不是很高、而且活动有送的一定的AI 平台的使用时间、而第三个则是不一样了、为了完成第三期、我大约使用了3天多的时间、包括整理记录。等所有完成后发现Notebook使用的费用比我预计的要高很多。此时就在想有没有办法把ModelArts-Lab AI实战营的内容,先以精简版本在本地测试、当实现了所有功能之后、再统一放到“ModelArts” 平台中来进行完整的结果验证

    这样做有二个好处:

        降低了成本:

        有了更加完成的学习记录

    以下信息为本地安装(tensorflow)的记录、希望可以帮助其它人提供一个引导、避免一些问题

    本地电脑配置(I7-16G-GT860M)

1: 查看官方指导手册(https://tensorflow.google.cn/install)、确认需要提前准备的软件 、并安装相关驱动 

    image.png

        按照顺序我们先安装 NVIDIA驱动 和GUDA工具包、Python环境使用Anacoda、所以SDK不需要安装 

        

2:下载"Anacoda"配置Python环境

    下载地址:https://www.anaconda.com/distribution/image.png

    有一点需要注意的、如上图所示、如果点击下载、则下载的是MacOS的包、需要先点击Window之后再点击对应版本的下载地址、本次环境使用Python3

    

    下载后执行安装(全部默认就可以了)、安装完成后先执行更新

        #LAB>codan update --all

      使用下面命令执行安装

        #LAB> conda install tensorflow-gpu


3:测试:

    首先把notebook环境创建

        创建notebook 登录的密码

        #LAB> jupyter notebook password

        创建notebook 配置文件

        #LAB> jupyter notebook --generate-config

        具备的配置可以参考附件(需要修改  c.NotebookApp.notebook_dir = 'D:\\40_Labcode'---需要修改为自己对应的目录)

    开打notebook笔记本

        #LAB> jupyter notebook

4: 使用示例测试、使用下面代码测试

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)


    5:查看 jupyter notebook日志信息

    

2019-07-21 22:12:00.616100: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2

2019-07-21 22:12:01.798520: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1433] Found device 0 with properties:

name: GeForce GTX 860M major: 5 minor: 0 memoryClockRate(GHz): 1.0975

pciBusID: 0000:01:00.0

totalMemory: 2.00GiB freeMemory: 1.64GiB

    如果上图所示、显示了GPU上关信息、则说明本地环境已经部署完成。

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。