Create an op on tensorflow; 在tensorflow 1.7/2.0 中创建一个 Op操作
2024-09-01 23:28:34
最近项目,需要创建一个 tensorflow 的一个自定义操作,用来加速tensorflow的处理效果;下面对创建过程中,遇到的问题和资源进行简要记录,进行备忘:
OP 创建 参考链接:
https://www.tensorflow.org/guide/create_op (官方教程)
https://github.com/tensorflow/custom-op (官方模板,看完上面的教程,使用该模板就可以很方便得在docker 容器中进行尝试构建;较为推荐)
何时定义一个新的OP:
- 现有的operation 组合不出来需要的OP;
- 现有的operation 组合出来的OP 十分低效;
- 你想要融合一些操作来提高效率;
保持更新,转载请注明出处;更多内容请关注 cnblogs.com/xuyaowen;
安装测试docker(用于gpu环境docker测试):
docker pull tensorflow/tensorflow:custom-op-gpu-ubuntu16
docker run --runtime=nvidia --privileged -it -v ${PWD}:/working_dir -w /working_dir tensorflow/tensorflow:custom-op-gpu-ubuntu16
安装测试docker(用于cpu环境docker测试):
docker pull tensorflow/tensorflow:custom-op-ubuntu16
docker run -it -v ${PWD}:/working_dir -w /working_dir tensorflow/tensorflow:custom-op-ubuntu16
docker run -it tensorflow/tensorflow:custom-op-ubuntu16 /bin/bash
使用清华镜像临时下载pip安装包:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
最新文章
- Core Java 总结(异常类问题)
- Reset Identity Column Value in SQL Server (Identity Reset)
- oracle 中数据库完全导入导出:cmd命令行模式(转载)
- IDE:Eclipse查看Servlet源码
- Eclipse配置Lifery SDK步骤与错误解决。
- Storm可靠性实例解析——ack机制
- Linux常见练习题
- [CareerCup] 17.11 Rand7 and Rand5 随机生成数字
- iOS开发UI篇—直接使用UITableView Controller
- poj2774 后缀数组 求最长公共子串
- [转]Calling Web Service Functions Asynchronously from a Web Page 异步调用WebServices
- 用HTML5构建一个流程图绘制工具
- canvas-js贝塞尔曲代码在线生成工具
- libiconv的静态编译
- C语言之结构体
- wcf类库及宿主
- Vue 爬坑之路(十一)—— 基于 Nuxt.js 实现服务端渲染(SSR)
- for循环 &;&; for-each
- PHP api接口和SQL数据库关联
- 学习笔记: 反射应用、原理,完成扩展,emit动态代码