珊瑚开发板中如何安装PyTorch

Question

我试图在PyTorch中安装珊瑚发展委员会。我找不到任何关于安装的在线参考。

• 我可以在中安装PyTorch移动设备(以前的Lite)吗？
• 架构支持安装吗？

Answer 1

我不是一个火把专家，但我不相信他们目前没有任何版本的arm平台。尽管人们已经能够从源头上构建它，并让它在像jetson这样的arm平台上工作。这可能是您可以遵循或基于以确定安装的最佳指南：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

FYI，我想提一下，edgetpu是专门用于处理tflite张量的。在进行安装之前，我只想确保您知道这一点，并意识到在使用pytorch时没有任何性能提升。