Ascend Pytorch Dockerfile Repository

This folder hosts the Dockerfile to build docker images with various platforms.

Build TorchNPU from Docker container

Clone TorchNPU

git clone https://gitcode.com/ascend/pytorch.git --depth 1

Build docker image

cd pytorch/ci/docker/{arch} # {arch} for X86 or ARM
docker build -t manylinux-builder:v1 .

If you want to configure the environment of LCOV, please build docker image like this:

cd pytorch/ci/docker/{arch} # {arch} for X86 or ARM
docker build -t manylinux-builder:v1 --build-arg CONFIG_FOR_LCOV=1 .

Enter docker Container

docker run -it -v /{code_path}/pytorch:/home/pytorch manylinux-builder:v1 bash
# {code_path} is the torch_npu source code path

Compile TorchNPU

Take Python 3.9 as an example

cd /home/pytorch
bash ci/build.sh --python=3.9