LibTorch使用 C++部署

pytorch的c++版本越來越好用了,我是從1.0版本用到1.3.1版本的,在開始時候很多pytorch的代碼用JIT script會出錯,然后就是很漫長的Debug,而且需要用到很多底層的libtorch C++ API去實現功能。
前兩天嘗試了1.3.1版本的,簡直不敢相信質的飛躍,基本上pytorch庫實現的python代碼只需要很少的修改就可以直接追蹤出模型。
而且早前編譯起來各種bug也沒了……猶記得很久以前編譯tensorflow,相比較一下,pytorch依賴特別少,基本上ubuntu桌面版裸機就行。
而且可以指定cuda版本去進行編譯,對于1.3的torch我嘗試發現最低是9.0的cuda版本
編譯指令很短的一條就可以:
CUDA_HOME=/usr/local/cuda-9.0/ python ../tools/build_libtorch.py

可以看出來pytorch下了大功夫進行部署上的優化
最后說一句,pytorch真香

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。