在部署AI模型的时候用到了库,直接从官网下载后发现与现有环境不兼容,执行发现缺失相应的GLIBC版本,因此选择下载源码重新编译libtorch。这里记录一下编译过程。
参照官网编译教程,主要分为以下几步:
anaconda3的安装教程可以参考: 安装anconda3
安装完后创建一个环境,我这里采用的python版本是3.7.4,所以执行以下命令
创建完成后,进入该环境,
安装相关依赖
克隆下来的是最新版本,执行查看分支,然后切换到想要的分支,
注意:如果下载比较慢或者报错,可以在pytorch目录下查看文件, 切换分支后先把里面网址替换为github加速插件的地址, 然后再执行git submodule sync 和 后面的命令
- 加速
以库为例
用export或者cmake-gui设置编译选项,如:
在分支目录下新建一个build文件夹,进入文件夹后执行以下命令,完成后在build目录下可以找到生成的结果。
新建文件夹
- 复制pytorch下的torch/include 到libtorch 目录
- 复制build/lib下编译好的库文件到 libtorch/lib
比如使用已有的MKL库,则可以修改目录下的文件,在里面指定你的库文件。
如果BLAS编译的时候设置为Eigen(默认为MKL),需要进行以下修改, 更多细节参考#32407
到此这篇libtorch 编译(libtorch编译)的文章就介绍到这了,更多相关内容请继续浏览下面的相关 推荐文章,希望大家都能在编程的领域有一番成就!版权声明:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若内容造成侵权、违法违规、事实不符,请将相关资料发送至xkadmin@xkablog.com进行投诉反馈,一经查实,立即处理!
转载请注明出处,原文链接:https://www.xkablog.com/rfx/74023.html