Aarch64架构:xavier安装torch-gpu

1.查看xavier安装的jetpack版本:

sudo apt show nvidia-jetpack
 一般情况下都是462的。

2.查看cuda版本:

cat /usr/local/cuda/version.txt
3.查看cudnn版本:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
4.1  torch-gpu-aarch64安装方式一:

参考nvidia相关网站 Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

可以在其中看到如下步骤:

 注意这个链接是jetpack-50版本/torch-1.12/python38版本的,可能不太匹配。

Go to->  https://developer.download.nvidia.cn/compute/redist/jp/ 可以看到其他版本的torch


这个版本是torch-1.11/python36的,且jetpack版本为461,和我们的 462版本很接近,可以安装。

剩下的安装过程参考官网第2条:

 安装完成后,不见得可以使用。在python 环境中 import torch 可能会出各种各样的问题。

4.2 torch-gpu-aarch64安装方式二

参考另一个网站:

PyTorch for Jetson - version 1.11 now available - Jetson Nano - NVIDIA Developer Forums

 这个里面torch版本较多,但是支持的语言只有python36和python27。

安装过程参考页面中Installation

4.3 torch-gpu-aarch64安装方式三

这个网站也提供了一些arm64/aarch64 架构的torch-gpu

Jetson Zoo - eLinux.org

5.torchvision安装

注意torchvision版本要和torch版本一一对应,这个链接可以看到对应的版本号:

GitHub - pytorch/vision at release/0.11

pip安装会出问题,推荐源码编译的方式安装。

#下载相关依赖
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
 
#下载0.x.0版本的torchvison
git clone -b v0.10.0 https://github.com/pytorch/vision.git vision-0.10.0
 
#源码编译
cd torchvision
export BUILD_VERSION=0.x.0  
#激活python环境,如果有anaconda的话
conda activate env
python3 setup.py install --user
6.aarch64架构上的一些疑难杂症,和解决办法

 6.1如果在xavier平台上,运行深度学习算法出现Core dump.则使用下一句

export OPENBLAS_CORETYPE=ARMV8
6.2如何在aarch64架构上安装点云处理软件ceres

sudo apt-get install liblapack-dev libsuitesparse-dev libcxsparse3 libgflags-dev libgoogle-glog-dev libgtest-dev
6.3 如何在aarch64架构上安装点云处理软件 Livox ROS Driver

#livox驱动地址:https://github.com/Livox-SDK/livox_ros_driver
 
sudo apt install cmake
sudo apt-get install gcc-aarch64-linux-gnu g++-aarch64-linux-gnu
 
git clone https://github.com/Livox-SDK/Livox-SDK.git
cd Livox-SDK
cd build && cmake .. -DCMAKE_SYSTEM_NAME=Linux -DCMAKE_C_COMPILER=aarch64-linux-gnu-gcc -DCMAKE_CXX_COMPILER=aarch64-linux-gnu-g++
make
sudo make install

  • 4
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值