当前位置:   article > 正文

Aarch64架构:xavier安装torch-gpu_loongarch64安装torch

loongarch64安装torch

Aarch64架构:xavier安装torch-gpu

1、查看 xavier 安装版本:

sudo apt show nvidia-jetpack
  • 1

2、查看 cuda 版本:

cat /usr/local/cuda/version.txt
  • 1

3、查看 cudnn 版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
  • 1

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

6.aarch64架构上的一些疑难杂症,和解决办法
6.1如果在xavier平台上,运行深度学习算法出现Core dump.则使用下一句

export OPENBLAS_CORETYPE=ARMV8
  • 1

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

sudo apt-get install liblapack-dev libsuitesparse-dev libcxsparse3 libgflags-dev libgoogle-glog-dev libgtest-dev
  • 1

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/336270
推荐阅读
相关标签
  

闽ICP备14008679号