I am trying to run Pytorch on my Xavier and running into the following issue which creates a segfault.
I have installed the wheel from the distribution provided on this link.
Failed to load Python extension for LZ4 support. LZ4 compression will not be available.
Loading data from:
…/tests/data/just_velodyne_points_short.bag
Loading scan: 3
Initializing network
[ 41 1200 1200]
Fatal Python error: Segmentation faultCurrent thread 0x0000007f823e2010 (most recent call first):
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 423 in convert
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 223 in _apply
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 201 in _apply
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 201 in _apply
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 201 in _apply
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 425 in to
File “/home/nvidia/Projects/lidar/src/map_processing/src/map_processing/second_utils.py”, line 159 in build_network
File “/home/nvidia/Projects/lidar/src/map_processing/src/map_processing/second_utils.py”, line 147 in init
File “xavier_test_script.py”, line 87 in
[1] 8864 segmentation fault (core dumped) python3 xavier_test_script.py -b …/tests/data/just_velodyne_points_short.bag
Here is the pip3 show torch torchvision
Name: torch
Version: 1.4.0
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /home/nvidia/.local/lib/python3.6/site-packages
Requires:
---
Name: torchvision
Version: 0.5.0a0
Summary: image and video datasets and models for torch deep learning
Home-page: https://github.com/pytorch/vision
Author: PyTorch Core Team
Author-email: soumith@pytorch.org
License: BSD
Location: /usr/local/lib/python3.6/dist-packages/torchvision-0.5.0a0-py3.6-linux-aarch64.egg
Requires: numpy, six, torch, pillow