Jetson Nano PyTorch Vision v0.7.0-rc2 Installation Error

I am trying to install Pytorch and Pytorch Vision as follows

wget https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl -O torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

pip3 install numpy torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

git clone --branch v0.7.0-rc2 GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision torchvision

sudo python3 setup.py install

It shows error message “Segmentation fault”

and not installing pytorch vision

my environment is
CUDA 10.2
cuDNN version: 8000

Torch 1.6.0 is installed properly
But i couldn’t install pytorch vision

Could you help me to install pytorch vision?

Hi,

It looks like the --branch v0.7.0 is somehow non-working.
Would you mind to checkout the branch on the folder directly?

$ git clone --branch v0.7.0-rc2 https://github.com/pytorch/vision torchvision 
$ cd torchvision
$ git checkout v0.7.0-rc2          # add this line
$ sudo python3 setup.py install
$ sudo pip3 install 'pillow<7'

Thanks.

I have executed the same steps as you mentioned. The same Segmentation Fault message is displaying when i install torch vision. Please help me to resolve this issue.

Refer the my environment

image

Hmm, not sure why you get the segfault while building torchvision, since you can import torch ok.

The l4t-pytorch and l4t-ml containers already include PyTorch and torchvision pre-installed - could you try using one of those?