深度學習之pytorch(三) C++調用

玩深度學習,個人覺得基於anaconda的python適合開發與測試,C++適合實際的工程部署!而pytorch官方有編譯好的libtorch,特別方便,適合於我這樣的伸手黨和手殘黨(win10下編譯tensorflow編譯了好久都沒通過,好憂傷!)好了,話不多說進入正題!

工程環境:

1.win10;

2.anaconda+pytorch1.0;

3.vs2015

4.Cmake

 

一、下載libtorch

進入pytorch官方網址https://pytorch.org/get-started/locally/,個人使用的是cpu版本,選擇如下圖所示:

下載好了放在無中文路徑的文件夾下(個人使用的是release版本)

二、Cmake配置

首先創建一個文件夾,存放主程序example.cpp還有CMakeLists.txt,然後再創建一個build的空文件夾,之後編譯好的文件都存放在build文件夾裏頭(照着教程來的,但是發現build文件夾內實際沒有工程文件,反而是根目錄文件夾內是編譯好的文件,即CMakeLists.txt所在文件夾,有知道的大佬煩請告知下原因)。cmake編譯文件夾目錄如下所示:

其中CMakeLists.txt內容如下所示:

#cmake的最低版本要求
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
#example爲創建的工程名字
project(example)
#添加的pytorch的附加庫
find_package(Torch REQUIRED)
#編譯的相應cpp文件
add_executable(example-app example-app.cpp)
target_link_libraries(example-app “${TORCH_LIBRARIES}”)
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)

其中example.cpp內容如下所示(僅作調用libtorch並生成隨機tensor):

#include <torch/torch.h>
#include <iostream>





int main() {
  torch::Tensor tensor = torch::rand({5, 3});
  std::cout << tensor << std::endl;
  system("pause");
}

三、Cmake編譯

camke編譯比較簡單,在根目錄文件夾下打開power shell(即CMakeLists.txt所在文件夾),輸入

cmake -DCMAKE_PREFIX_PATH=E:\lib\pytorch\libtorch 

DCMAKE_PREFIX_PATH=libtorch絕對路徑,就是剛纔下載的libtorch文件解壓後的根目錄。

編譯成功後會出現以下結果(並且剛纔相應的文件會出現VS工程項目):

四、結果運行

用vs2015打開如下文件

1.打開工程文件後,修改配置管理器爲X64,以及將example項目設置爲啓動項目(如果不會,請自行百度,這是vs的基本操作)。2.打開example項目的屬性->鏈接器->輸入->附加依賴項,刪除其中引號內的所有內容,否則會出現找不到torch.lib的錯誤提示;

3.接上述步驟,->命令行->其他選項,將其中的X86更改爲X64;

運行example項目會提示找不到相應的lib文件,此時繼續下面步驟,

4.將之前解壓的libtorch文件(\path\libtorch\lib)下的lib文件夾內的所有內容(內容如下圖所示)拷貝至\path\example-app\x64\Release文件夾下,

此時項目運行成功,結果如下圖所示:

 

注意:

1.CamkeLists.txt,主要爲設置相應的項目屬性以及需要編譯的附加庫,但是不明白爲什麼還需要將相應的libtorch庫文件拷貝至當前項目環境下,難道是因爲沒有在環境變量裏設置libtorch的變量路徑嗎?下次有空再測試;

2.libtorch解壓路徑不要太深,貌似是因爲Cmake路徑不能超過一千個字符,所以路徑儘量淺一點,還有就是作爲碼農,不要有中文路徑,畢竟外國人基本不用中文,在碼農世界中文是個蛋疼的玩意,會導致很多莫名其妙的錯誤,避免坑就戒掉中文路徑吧!

3.Camke是個好東西,可以多學習學習,下面有相應的CamkeLists.txt的參考博客,覺得都是大牛啊!

 

參考文獻:

1.https://blog.csdn.net/adyf3/article/details/85003893

2.https://blog.csdn.net/qq_37761077/article/details/88750711

3.https://blog.csdn.net/linxigjs/article/details/79864033

4.https://pytorch.org/cppdocs/installing.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章