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真香