JetsonNano(JetPack4.4)でPyTorchを動かす

nvidia-image-logo 開発関連
スポンサーリンク
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
タイトルとURLをコピーしました