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