Introduction

This document describes how to install PyTorch on Kunpeng servers and verify its basic functions.

PyTorch provides flexible and easy-to-use tensor computation and automatic differentiation capabilities based on its core features: dynamic computational graph (Define by Run) and immediate execution mode. Deploying PyTorch on Kunpeng servers allows for operator-level deep optimization for Ascend NPUs and Kunpeng processors through Kunpeng PyTorch extension (KPEX). Additionally, the collaboration between the KML math library and BiSheng compiler significantly accelerates intensive computations such as GEMM at the fully connected layer.