jetson agx xavierにPyTorchをインストールする
pipからpytorchをインストールしようとすると,このようなエラーがでちゃいました...
Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-7PRgOy/torch/
ARM用にビルドされたバイナリを使う必要がある
こんな手順でできるそうです. これは公式がサポートしてくれています. 1.7.0をインストールする手順を示していますが,他のバージョンのインストールも可能です. (URLとかは先程のリンクにあります!)
pip3 install cython wget https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl mv cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl torch-1.7.0-cp36-cp36m-linux_aarch64.whl pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl sudo apt install libopenblas-dev
cythonをインストールせずに行うとこういうエラーがでます
ModuleNotFoundError: No module named 'Cython'
Command "/usr/bin/python3 -u -c "import setuptools, tokenize;__file__='/tmp/pip-build-3caqz9c0/numpy/setup.py';f=getattr(tokenize, 'open', open)(__file__);code=f.read().replace('\r\n', '\n');f.close();exec(compile(code, __file__, 'exec'))" install --record /tmp/pip-qk46wyli-record/install-record.txt --single-version-externally-managed --compile --user --prefix=" failed with error code 1 in /tmp/pip-build-3caqz9c0/numpy/
libopenblas-devをインストールせずにtorchをimportするとこんなエラーもでます
libopenblas.so.0: cannot open shared object file: No such file or directory
追記: torchvisionのインストール
sudo apt install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev git clone --branch v0.8.1 https://github.com/pytorch/vision torchvision cd torchvision export BUILD_VERSION=0.8.1 pip3 install --user setuptools python3 setup.py install --user