torch-1.4をいれようとしたらエラーがでたのでメモ。。。
↓発生したエラー
ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory
解決方法
JetPack4.4からはv1.6以上じゃないとダメらしい。↓インストール手順です
$ wget https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl -O torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl
$ apt-get install python3-pip libopenblas-base libopenmpi-dev
$ pip3 install Cython numpy
$ pip3 install torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl