天津大学
基于华为自研NPU Ascend 910的TensorFlow 1.x训练脚本迁移和使能混合精度记录
墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年，写一次发全网
我也要 关注 + 加入社区基于华为自研NPU Ascend 910的TensorFlow 1.x训练脚本迁移和使能混合精度记录
简介基于 TesorFlow 1.x 以 Sess.run 形式搭建入门级——手写数字分类网络，并迁移到 华为自研NPU Ascend 910，同时使能混合精度。硬件介绍华为自研 NPU Ascend 910，即昇腾 910 AI 处理器（简称NPU），根据官方介绍，是在2019年发布的人工智能（AI）专用的神经网络处理器，其算力高达256T，最新款算力高达310T，是业界主流芯片算力的2倍。当前业界大多数训练脚本基于 TensorFlow 的 Python API 开发，默认运行在 CPU/GPU/
