在NVIDIA Jetson TX2上源碼編譯並使用Pytorch

前言 TX2作爲一個嵌入式平臺的深度學習端,具備不錯的GPU性能,我們可以發現TX2的GPU的計算能力是6.2。這意味着TX2對半精度運算有着良好的支持,我們完全可以在桌面端訓練好模型,然後移植到TX2上利用半精度運行進行推理,這樣可以達到生產落地的效果。 就算本篇文章主要展示瞭如何在TX2中源碼編譯Pytorch-1.0。 首先我們需要一個相對純淨的jetpack系統,3.2-3.3版本(最新的
相關文章
相關標籤/搜索