玩深度學習,個人覺得基於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