admin管理员组

文章数量:1278882

I'm trying to install PyTorch with CUDA support on a Jetson Nano. It's running the Ubuntu 18.04 that the jetson-nano-jp461-sd-card-image image from Nvidia put on it with Python 3.6.9, CUDA 10.2, and JetPack 4.6.6. I tried:

pip3 install torch --index-url

but this gives me the CPU-only version, and I need CUDA support for the Nano's GPU. I also tried

git clone --recursive --branch v1.8.0

and installing, but it is really, really, slow.

Is there a pre-built wheel available for this specific configuration? Many thanks in advance.

I'm trying to install PyTorch with CUDA support on a Jetson Nano. It's running the Ubuntu 18.04 that the jetson-nano-jp461-sd-card-image image from Nvidia put on it with Python 3.6.9, CUDA 10.2, and JetPack 4.6.6. I tried:

pip3 install torch --index-url https://download.pytorch./whl/cpu

but this gives me the CPU-only version, and I need CUDA support for the Nano's GPU. I also tried

git clone --recursive --branch v1.8.0 https://github/pytorch/pytorch

and installing, but it is really, really, slow.

Is there a pre-built wheel available for this specific configuration? Many thanks in advance.

  • pytorch
  • nvidia-jetson-nano
Share Improve this question asked Feb 24 at 2:56 DribblerDribbler 4,70110 gold badges37 silver badges57 bronze badges
Add a comment  | 

1 Answer 1

Reset to default 0

For those who may be searching, this worked for me:

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev

pip3 install 'Cython<3'

pip3 install numpy

wget https://nvidia.box/shared/static/fjtbno0vpo676a25cgvuqc1wty0fkkg6.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl

sudo pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl

本文标签： Is there a prebuilt wheel for PyTorch for Jetson NanoStack Overflow

版权声明：本文标题：Is there a pre-built wheel for PyTorch for Jetson Nano? - Stack Overflow 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.betaflare.com/web/1741296161a2370830.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。