pytorch拓展-cpp extension

日期:2020-6-9
折腾了好久,总算把拓展这件事情做完了一次,不易。
这篇文章写写这个过程中自己所遇到的困难和解决的方法。
首先是文档:建议大家不用看太多的博客,照着官方的tutorial走一遍,会有很多的收获
连接:https://pytorch.org/tutorials/advanced/cpp_frontend.html
pytorch拓展-cpp extension_第1张图片

配置libtorch

libtorch是torch的c++库,配置他的作用在于这样可以用cpp的来实现pytorch相同的功能,而且可以结合cuda一起使用,这样的话可以充分利用计算资源,实现自己想要达成的效果,这在现实中的神经网络项目的论文实现中非常重要。
pytorch拓展-cpp extension_第2张图片
设置好环境变量,每次建立新项目的时候需要调的东西:
1.包含目录
2.库目录
3.连接器/输入/附加依赖项
这里跟OpenGL是一样的

pip需要装的库

在这过程中我装了nijia和pybind11(cpp宏定义调用)他们的作用在于构建Pytorch和cpp的联系

setup遇到的问题

写好setup.py之后,需要在命令行执行
1.cd …更变目录
2.输入:python setup.py install
成功的情况下的结果是这样的
在这里插入图片描述
中间我遇到的问题是找不到cl.exe(编译cpp的关键)
我们这是要需要做的事情是找到这个文件所在的位置,然后添加进环境变量中系统变量的PATH里。
可以参考这篇文章:
https://blog.csdn.net/fanre/article/details/104340318

以上就是我的整个历程,说起来好像没有啥,但确实花费了不少时间,希望自学的大家大家看了我这篇文后能少走弯路吧。

你可能感兴趣的:(pytorch初学笔记,人工智能,c++,python)