pytorch-aarch64 icon indicating copy to clipboard operation
pytorch-aarch64 copied to clipboard

support the 'RK3566' or 'RK3568' soc?

Open timiil opened this issue 4 years ago • 4 comments

hello, as title say,the Rockchip had release rk3566 soc (4 * core A55, 0.8 TOPS NPU, Ubuntu 18.04),they also said that it is support caffe,ONNX,pytorch ... framworks.

so,do we have any plan to this SOC on this project? tks.

timiil avatar Aug 30 '21 23:08 timiil

Hi.

This project aims to support all ARMv8 64bit processors, so the chips you mentioned should already be supported.

As for the NPU, PyTorch itself does not support GPUs other than Nvidia and AMD.

KumaTea avatar Aug 31 '21 09:08 KumaTea

Hi.

This project aims to support all ARMv8 64bit processors, so the chips you mentioned should already be supported.

As for the NPU, PyTorch itself does not support GPUs other than Nvidia and AMD.

thanks for your reply. you mean the soc may be already supported by this project , but without the NPU computing enhancement?

timiil avatar Aug 31 '21 14:08 timiil

Hi.

This project aims to support all ARMv8 64bit processors, so the chips you mentioned should already be supported.

As for the NPU, PyTorch itself does not support GPUs other than Nvidia and AMD.

thanks for your reply. you mean the soc may be already supported by this project , but without the NPU computing enhancement?

timiil avatar Aug 31 '21 14:08 timiil

thanks for your reply. you mean the soc may be already supported by this project , but without the NPU computing enhancement?

Yes. However some boards may contain proprietary NPU chips (SPR2801S for example), which may gain significantly performance improvement after converting PyTorch model to their proprietary formats.

KumaTea avatar Aug 31 '21 18:08 KumaTea