Jetson AGX ORIN 配置 FGVC-PIM 神经网络(包含 arm64 下面 torch 和 torchvision 配置内容)

本文主要是介绍Jetson AGX ORIN 配置 FGVC-PIM 神经网络(包含 arm64 下面 torch 和 torchvision 配置内容),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

Jetson AGX ORIN 配置 FGVC-PIM 神经网络

文章目录

  • Jetson AGX ORIN 配置 FGVC-PIM 神经网络
    • 配置 ORIN 环境
    • 创建 FGVC-PIM 虚拟环境
      • 安装 PyTorch
      • 安装 torchvision
      • 安装其他依赖包

配置 ORIN 环境

首先先配置 ORIN 的环境,可以参考这个链接:

Jetson AGX ORIN 初始化&配置CUDA&Anaconda&带CUDA的OpenCV

创建 FGVC-PIM 虚拟环境

终端输入命令:

conda create -n fgvc python=3.8
conda activate fgvc

即可创建名为 fgvc 的虚拟环境,之后需要向环境中添加需要的安装包。python 一定要安装 38 版本,因为安装 PyTorch 的时候需要对照版本进行安装。如果直接使用这个命令进行安装 pip install torch torchvision torchaudio,则会出现无法调用 CUDA 的问题。

安装 PyTorch

安装 PyTorch 之前先要查看一下 ORIN 的版本信息:

  • L4T:35.3.1
  • Jatpack:5.1.1

然后在官网里面下载安装包。网址如下:
PyTorch for Jetson

我的 ORIN 信息如上,选择对应的 python=3.8 系列,最终确定 PyTorch 版本为 1.12.0,进行下载。
如图所示:

在这里插入图片描述然后激活对应的虚拟环境,进行包的安装。

conda activate fgvc
pip install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl 

显示成功安装后,进行 python 界面,测试是否可以导入,以及 CUDA 是否可用。出现如下结果说明一切正常,可以进行下一步 torchvision 安装。如下所示:

(fgvc) abc@ubuntu:~$ python
Python 3.8.19 (default, Mar 20 2024, 19:53:40) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.12.0a0+2c916ef.nv22.3'
>>> torch.cuda.is_available()
True
>>> 

中间可能会出现一个小问题,显示没有 libopenblas-dev 东西,用如下命令进行安装即可解决:

sudo apt-get install libopenblas-dev

安装 torchvision

安装之前先运行以下命令:

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

在环境变量里面添加内容:

export CUDA_HOME=/usr/local/cuda-11.4
# 然后更新
source ~/.bashrc

再看这个对照表

在这里插入图片描述

使用 git clone 下载 torchvision 的对应安装代码。

git clone --branch v0.13.0 https://github.com/pytorch/vision torchvision

然后在虚拟环境终端里面输入:

cd torchvision
export BUILD_VERSION=v0.13.0

然后再输入:

python3 setup.py install --user

之后进行调试,还是会有问题,但是具体基本都是一些库需要下载,进行下载即可。
可以参考这个链接:(里面的 export 不要加 v,不然版本会不匹配)
jetson agx orin 的pytorch、torchvision、tensorrt安装最全教程

# 下载 torchvision 时候报错 需要 numpy
Downloading https://files.pythonhosted.org/packages/51/fe/e4dab289c176ea4e13f97f11f281cc22d4a3b0add9883406db62d4f94d65/numpy-2.0.0b1.tar.gz#sha256=e0bb33a37d0d0b9a19cd41a093877f830e06bd4d989341b9792896cf08e629f7
Best match: numpy 2.0.0b1
Processing numpy-2.0.0b1.tar.gz
error: Couldn't find a setup script in /tmp/easy_install-di_cher0/numpy-2.0.0b1.tar.gz
# 下载 numpy 和 numpy 需要的其他库
(fgvc) abc@ubuntu:~/torchvision$ pip install certifi idna charset-normalizer numpy urllib3
# 参考上面的链接提前安装一些 torchvision 的依赖库

再运行之后问题解除
然后在 python 里面导入 torchvision,看是否有问题,结果如下,能出来 ‘v0.13.0’ 则可以继续安装其他依赖包:

(fgvc) be@ubuntu:~/torchvision$ python
Python 3.8.19 (default, Mar 20 2024, 19:53:40) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
/home/be/torchvision/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: warn(f"Failed to load image Python extension: {e}")
/home/be/torchvision/torchvision/__init__.py:28: UserWarning: You are importing torchvision within its own root folder (/home/be/torchvision). This is not expected to work and may give errors. Please exit the torchvision project source and relaunch your python interpreter.warnings.warn(message.format(os.getcwd()))
>>> torch.__version__
'1.12.0a0+2c916ef.nv22.3'
>>> torchvision.__version__
'0.13.0'
>>> 

安装其他依赖包

进入 pycharm,并在 Setting 设置好代码的虚拟环境之后,根据缺少的库的内容,使用 pip install 命令下载库。
需要的库及其命令如下,不需要找了,可以直接下载。

pip install numpy pandas matplotlib wandb psutil
pip install opencv-python
pip install scipy scikit-learn

如果下载中断,可以使用镜像源:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pandas

或者使用参数 –default-timeout

pip install --default-timeout=600 pandas

然后再运行训练数据集的命令。
提前下载权重文件:
swin_large_patch4_window12_384_22k.pth,网址如下:
https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth

下载好之后,将权重文件放置在这个路径下:
/home/abc/.cache/torch/hub/checkpoints
然后就发现可以训练了(这个不打算训练,所以 Ctrl + c 强行中断了),配置结束!
在这里插入图片描述

这篇关于Jetson AGX ORIN 配置 FGVC-PIM 神经网络(包含 arm64 下面 torch 和 torchvision 配置内容)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/848700

相关文章

windos server2022的配置故障转移服务的图文教程

《windosserver2022的配置故障转移服务的图文教程》本文主要介绍了windosserver2022的配置故障转移服务的图文教程,以确保服务和应用程序的连续性和可用性,文中通过图文介绍的非... 目录准备环境:步骤故障转移群集是 Windows Server 2022 中提供的一种功能,用于在多个

windos server2022里的DFS配置的实现

《windosserver2022里的DFS配置的实现》DFS是WindowsServer操作系统提供的一种功能,用于在多台服务器上集中管理共享文件夹和文件的分布式存储解决方案,本文就来介绍一下wi... 目录什么是DFS?优势:应用场景:DFS配置步骤什么是DFS?DFS指的是分布式文件系统(Distr

关于Maven中pom.xml文件配置详解

《关于Maven中pom.xml文件配置详解》pom.xml是Maven项目的核心配置文件,它描述了项目的结构、依赖关系、构建配置等信息,通过合理配置pom.xml,可以提高项目的可维护性和构建效率... 目录1. POM文件的基本结构1.1 项目基本信息2. 项目属性2.1 引用属性3. 项目依赖4. 构

龙蜥操作系统Anolis OS-23.x安装配置图解教程(保姆级)

《龙蜥操作系统AnolisOS-23.x安装配置图解教程(保姆级)》:本文主要介绍了安装和配置AnolisOS23.2系统,包括分区、软件选择、设置root密码、网络配置、主机名设置和禁用SELinux的步骤,详细内容请阅读本文,希望能对你有所帮助... ‌AnolisOS‌是由阿里云推出的开源操作系统,旨

mysql-8.0.30压缩包版安装和配置MySQL环境过程

《mysql-8.0.30压缩包版安装和配置MySQL环境过程》该文章介绍了如何在Windows系统中下载、安装和配置MySQL数据库,包括下载地址、解压文件、创建和配置my.ini文件、设置环境变量... 目录压缩包安装配置下载配置环境变量下载和初始化总结压缩包安装配置下载下载地址:https://d

gradle安装和环境配置全过程

《gradle安装和环境配置全过程》本文介绍了如何安装和配置Gradle环境,包括下载Gradle、配置环境变量、测试Gradle以及在IntelliJIDEA中配置Gradle... 目录gradle安装和环境配置1 下载GRADLE2 环境变量配置3 测试gradle4 设置gradle初始化文件5 i

SpringCloud配置动态更新原理解析

《SpringCloud配置动态更新原理解析》在微服务架构的浩瀚星海中,服务配置的动态更新如同魔法一般,能够让应用在不重启的情况下,实时响应配置的变更,SpringCloud作为微服务架构中的佼佼者,... 目录一、SpringBoot、Cloud配置的读取二、SpringCloud配置动态刷新三、更新@R

MySQL中my.ini文件的基础配置和优化配置方式

《MySQL中my.ini文件的基础配置和优化配置方式》文章讨论了数据库异步同步的优化思路,包括三个主要方面:幂等性、时序和延迟,作者还分享了MySQL配置文件的优化经验,并鼓励读者提供支持... 目录mysql my.ini文件的配置和优化配置优化思路MySQL配置文件优化总结MySQL my.ini文件

C#读取本地网络配置信息全攻略分享

《C#读取本地网络配置信息全攻略分享》在当今数字化时代,网络已深度融入我们生活与工作的方方面面,对于软件开发而言,掌握本地计算机的网络配置信息显得尤为关键,而在C#编程的世界里,我们又该如何巧妙地读取... 目录一、引言二、C# 读取本地网络配置信息的基础准备2.1 引入关键命名空间2.2 理解核心类与方法

最新版IDEA配置 Tomcat的详细过程

《最新版IDEA配置Tomcat的详细过程》本文介绍如何在IDEA中配置Tomcat服务器,并创建Web项目,首先检查Tomcat是否安装完成,然后在IDEA中创建Web项目并添加Web结构,接着,... 目录配置tomcat第一步,先给项目添加Web结构查看端口号配置tomcat    先检查自己的to