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
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?