如何在Coral Dev Board中安装PyTorch

Question

我试图在Coral Dev Board中安装PyTorch。我找不到安装的任何在线参考。

• 我可以在Coral Dev Board中安装PyTorch Mobile（以前为Lite）吗？
• 体系结构是否支持安装？

Answer 1

我不是pytorch的专家，但我不认为他们目前有任何适用于arm平台的版本。尽管人们已经能够从源代码上构建它，并使它在Jetson Nano等手臂平台上运行。这可能是您可以遵循或从中找出安装的最佳指南： https://gist.github.com/dusty-nv/ef2b372301c00c0a9d3203e42fd83426

我对其进行了少许修改以适合开发板，但尚未尝试：

# clone pyTorch repo
git clone http://github.com/pytorch/pytorch
cd pytorch
git submodule update --init --recursive

# install prereqs
sudo pip3 install -U setuptools
sudo pip3 install -r requirements.txt

# Develop Mode:
python3 setup.py install

仅供参考，我想提一下edgetpu专用于处理tflite张量。我只是想确保在安装之前知道这一点，并意识到使用pytorch API不会提高性能。