Ubuntu下vscode配置OpenCV以及Libtorch

opencv安装

  1. sudo apt-get update
  2. sudo apt-get install libopencv-dev

该方式安装的版本可能比较旧。

测试代码

   #include 
   #include 

   int main() {
       cv::Mat image = cv::imread("t.png");
       cv::imshow("Image", image);
       cv::waitKey(0);
       return 0;
   }

Ctrl + F5,然后选择g++ build active file 运行代码。

出现对‘cv::imread(cv::String const&, int)’未定义的引用

编辑task.json文件,在args参数下末尾添加
“-lopencv_core”,
“-lopencv_imgcodecs”,
“-lopencv_highgui”

手动安装指定版本opencv

  1. 官网下载指定版本的source代码,并解压到本地。

  2. 解压后执行make命令

    mkdir build
    cd build
    cmake ..
    make -j8
    sudo make install
    
  3. /etc/ld.so.conf.d/路径下创建任意一个.conf文件,把lib文件的路径写在里面,一般是 /usr/local/lib,然后执行 sudo ldconfig即可。(否则会报错:error while loading shared libraries: libopencv_core.so.3.4: cannot open shared object file)

Libtorch配置

  1. 下载对应版本并解压。历史版本

  2. 使用cmake进行构建。cmake内容如下:

    cmake_minimum_required(VERSION 2.6)
    
    project(libtorch_t)
    
    set(Torch_DIR /home/yang/Documents/libtorch/share/cmake/Torch)  #你解压的libtorch的绝对路径
    find_package(Torch REQUIRED)
    
    set(CMAKE_CXX_FLAGS "${CAMKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
    #set(CUDA_INCLUDE_DIRS "/usr/local/cuda/include")
    
    #main.cpp exe
    add_executable(libtorch_t libtorch_t.cpp)
    
    #link libtorch .a .so
    target_link_libraries(libtorch_t "${TORCH_LIBRARIES}")
    target_include_directories(libtorch_t PRIVATE CUDA_INCLUDE_DIRS)
    
    set_property(TARGET libtorch_t PROPERTY CXX_STANDARD 14)
    
  3. 编写libtorch_t测试代码

    #include 
    #include 
    
    int main() {
      torch::Tensor tensor = torch::rand({2, 3});
      std::cout << tensor << std::endl;
    }
    
  4. 运行run.sh 其内容如下:
    cmake .
    make
    ./libtorch_t

你可能感兴趣的:(OpenCV,ubuntu,ubuntu,vscode,opencv,libtorch)