第一次写这种这么长的 readme，介绍了如何在 pytorch 中用 CUDA 写自定义扩展的方法，对搞 nlp 或者 cv 的小伙伴们应该有所帮助，地址在这，how to write cuda extensions in pytorch，求 star^_^，顺便说一句 gboard 挺好用的啊...
这是一个专为移动设备优化的页面（即为了让你能够在 Google 搜索结果里秒开这个页面），如果你希望参与 V2EX 社区的讨论，你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区，是一个分享自己正在做的有趣事物、交流想法，可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.