第一次写这种这么长的 readme,介绍了如何在 pytorch 中用 CUDA 写自定义扩展的方法,对搞 nlp 或者 cv 的小伙伴们应该有所帮助,地址在这,how to write cuda extensions in pytorch,求 star^_^,顺便说一句 gboard 挺好用的啊...
1
zhuangjia 2018-03-16 11:12:18 +08:00
再提供一个中文版可能 star 会更多点?
|