Create an op on tensorflow; 在tensorflow 1.7/2.0 中创建一个 Op操作

xuyaowen

发布于 2020-12-30 11:26:19

7650

发布于 2020-12-30 11:26:19

文章被收录于专栏：XUYAOWEN的专栏

最近项目，需要创建一个 tensorflow 的一个自定义操作，用来加速tensorflow的处理效果；下面对创建过程中，遇到的问题和资源进行简要记录，进行备忘：

OP 创建参考链接：

https://github.com/tensorflow/custom-op （官方模板，看完上面的教程，使用该模板就可以很方便得在docker 容器中进行尝试构建；较为推荐）

何时定义一个新的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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019-11-04 ，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度