AArch64でPyTorch、ONNX、TVM

@Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった

TensorFlowを Raspberry Pi で動きますが、PyTorchはどうなんでしょうか？

頑張って、いろいろやってビルドすればできると思いますが、超めんどくさそう。

ということで、見つけました。

mathinf.eu

Debian Buster ARM64で動く

• onnx-1.5.0-cp37-cp37m-linux_aarch64.whl
• onnxruntime-0.5.0-cp37-cp37m-linux_aarch64.whl 
• topi-0.6.dev0-py3-none-any.whl 
• torch-1.3.0a0-cp37-cp37m-linux_aarch64.whl 
• torch_tvm-0.0.1-cp37-cp37m-linux_aarch64.whl  
• torchvision-0.5.0a0-cp37-cp37m-linux_aarch64.whl  
• tvm-0.6.dev0-cp37-cp37m-linux_aarch64.whl
• xgboost-0.90-cp37-cp37m-linux_aarch64.whl 

Torchだけでなく、TVM、ONNX、ONNXRuntimeもあります。

 PyTorch on the Raspberry Pi workshop というものの成果なんですかね？