本文主要是介绍Linux下libtorch导入VTM编码器,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1. libtorch下载
此处注意下载和pytorch版本一致的libtorch
wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.0.1%2Bcpu.zip
unzip 下载的libtorch压缩包
2. 修改CMakeLists.txt
- 首先在VTM编码器的文件夹下的
CMakeLists.txt
中,添加相对当前vtm文件夹路径的 libtorch头文件,库路径,如图
然后修改此CMakeLists.txt
中的c++编译器版本至c++14,如图
然后注释此CMakeLists.txt
中的警告,如图
- 继续修改VTM中目录
VTM/source/App/EncoderApp/
下的CMakeLists.txt
文件,给EncoderApp添加编译库路径,也就是把libtorch/lib/下的库添加进去,为了保证正常运行我就全添加了,如图,
3. 注释或者修改libtorch中与VTM编码器中的重命名
在目录~libtorch\include\c10\util
下的logging_is_not_google_glog.h
头文件中
注释掉重命名 CHECK
119 // Check for a given boolean condition.
120 // #define CHECK(condition) FATAL_IF(condition) << "Check failed: " #condition " "
4. 运行
可能运行还是会出现问题,多半是libtorch库路径的问题,把相关路径配置到用户目录下的 .bashrc
文件下,如图
保险起见也可以在 /etc/profile
文件下面配置此路径,并且把libtorch的库包复制到/usr/lib/
目录下
这篇关于Linux下libtorch导入VTM编码器的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!