当前位置:   article > 正文

torch &pytorch_pytorch底层用什么写的

pytorch底层用什么写的

pytorch  是python 版的torch,底层大部分是C++ 语言。依靠nvidia cudnn、Intel MKL和NNPACK来加速。

torch 是用lua语言写的。低层使用C语言写的。

因为pytorch的设计bug原因,在pytorch中使用.c.cu文件时时,需要首先编译一下.cu然后将编译好的.cu包含在路径中和.c一起再用cffi进行编译才能使用。

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/979971
推荐阅读
相关标签
  

闽ICP备14008679号