pytorch拓展-cpp extension

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

配置libtorch

libtorch是torch的c++库,配置他的作用在于这样可以用cpp的来实现pytorch相同的功能,而且可以结合cuda一起使用,这样的话可以充分利用计算资源,实现自己想要达成的效果,这在现实中的神经网络项目的论文实现中非常重要。
在这里插入图片描述
设置好环境变量,每次建立新项目的时候需要调的东西:
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

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

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值