Wiita

自分にとってのメモと, プログラミングに関する情報を発信していきます. サイト名の意味は特にありません.

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

www.syarudhi.net