在NVIDIA Jetson TX2上源码编译并使用Pytorch

前言 TX2作为一个嵌入式平台的深度学习端,具备不错的GPU性能,我们可以发现TX2的GPU的计算能力是6.2。这意味着TX2对半精度运算有着良好的支持,我们完全可以在桌面端训练好模型,然后移植到TX2上利用半精度运行进行推理,这样可以达到生产落地的效果。 就算本篇文章主要展示了如何在TX2中源码编译Pytorch-1.0。 首先我们需要一个相对纯净的jetpack系统,3.2-3.3版本(最新的
相关文章
相关标签/搜索