Xavier安装pytorch

1安装conda(miniforge)

NVIDIA Jetson Xavier的aarch64架构无法使用anaconda,可以使用miniforge替代.

下载:
miniforge的github地址
miniforge的.sh release下载地址

安装:

sh Miniforge-pypy3-4.8.3-4-Linux-aarch64.sh

使用基本同conda

conda activate XXX
conda deactivate
conda create -n open-mmlab python=3.6
conda remove -n open-mmlab --all
conda install 
conda uninstall 

2安装pytorch

1、创建环境

conda create -n open-mmlab python=3.6 -y
conda activate open-mmlab

2、下载安装
需要到Nvidia PyTorch for Jetson下载对应版本的pytorch和torchvision,此网站有详细安装说明,照着做即可:

  • 下载对应的版本(pytorch和CUDA对应关系)为使用mmdetection,我安装的版本为pytorch1.3.0+torchvision0.4.2(系统环境: jetpack4.2.3+L4T32.2.1+cuda10.0+cudnn7.3)
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
  • 安装成功pytorch之后(先别安torchvision),检查下是否安装成功:
python3
>>>import torch

如果报错:Illegal instruction(core dumped)
可以在每次运行Python指令前运行:export OPENBLAS_CORETYPE=ARMV8
也可以将export OPENBLAS_CORETYPE=ARMV8加入到“~/.bashrc”中

  • 安装torchvision
    (where 0.X.0 is the torchvision version,在网站上可以看到)
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.X.0 https://github.com/pytorch/vision torchvision   
cd torchvision
export BUILD_VERSION=0.X.0  
python3 setup.py install --user
cd ../  
pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

3安装mmdetection

以下均参考github
1、Install MMCV

pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu100/torch1.3.0/index.html

2、Install MMDetection.

pip install git+https://github.com/open-mmlab/mmdetection.git

3、Install MMDetection3D

git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip install -v -e .

可能错误

Building wheel for llvmlite (setup.py) ... error
RuntimeError: llvm-config failed executing, please point LLVM_CONFIG to the path for llvm-config

原因:没有安装LLVM

conda install llvm

依旧有错误

Be sure to set LLVM_CONFIG to the right executable path.

原因并没有找到LLVM路径,手动指定路径=/usr/lib/llvm-8/bin/llvm-config(每个人可能个不一样)

LLVM_CONFIG=/usr/lib/llvm-8/bin/llvm-config pip install -v -e .

4、运行

python demo/pcd_demo.py demo/kitti_000008.bin configs/second/hv_second_secfpn_6x8_80e_kitti-3d-car.py checkpoints/second.pth
python demo/pcd_demo.py demo/sunrgbd_000017.bin configs/votenet/votenet_8x8_scannet-3d-18class.py checkpoints/votenet.pth
python demo/pcd_demo.py demo/sunrgbd_000017.bin configs/h3dnet/h3dnet_3x8_scannet-3d-18class.py checkpoints/h3dnet.pth

votenet等许多模型会报错:The model and loaded state dict do not match exactly
但官方只有votenet的解决方式:转换下模型再用

python ./tools/model_converters/convert_votenet_checkpoints.py checkpoints/votenet.pth --out=checkpoints/vot.pth

报错…
1、

libgeos_c.so: cannot open shared object file: No such file or directory

解决:

conda install geos

2、

libstdc++.so.6: version `GLIBCXX_3.4.26' not found (required by /home/xavier/miniforge3/envs/open-mmlab/lib/./libgeos-3.9.1.so)

解决:

cd ~/miniforge3/envs/open-mmlab/lib
sudo cp libstdc++.so.6.0.28 /usr/lib/aarch64-linux-gnu
cd /usr/lib/aarch64-linux-gnu
sudo rm -f libstdc++.so.6
ln -s libstdc++.so.6.0.28 libstdc++.so.6

3、

 module 'pycocotools' has no attribute '__version__'

解决:

pip list | grep pycocotools
pip uninstall pycocotools
pip uninstall mmpycocotools
pip install mmpycocotools

4、

libgomp-d22c30c5.so.1.0.0: cannot allocate memory in static TLS block

解决:bashrc添加:

export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1

4安装Open3D

安装完成以上就可以运行mmdetection了,但是没有可视化,需要安装open3D,然而arrch64没有open3D,源码编译安装又是一堆坑…

  • 5
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值