jetson orin nano super 环境初始化

基本信息

硬件：jetson orin nano super

系统：Ubuntu 20.04

python yolo11环境初始化

安装下面所有python包

# 安装pip3
sudo apt-get install python3-pip
pip3 install --upgrade pip

# 安装必要的包
sudo pip3 install tqdm
sudo pip3 install thop
sudo pip3 install psutil
sudo pip3 install typeDict
sudo pip3 install lap==0.4.0
sudo pip3 install numpy==1.21

1、下载jetpack专用的Pytorch源码，下载：torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

2、然后前往GitHub仓库下载对应版本的torchversion：vision-0.16.0.zip

https://github.com/pytorch/vision

下载完并解压，然后进入该项目文件夹下，运行以下代码执行安装

sudo apt-get install libopenblas-dev
sudo python3 setup.py install

接下来配置环境变量，输入以下代码

vim ~/.bashrc

进入文件，在文件最下方复制粘贴进以下代码

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_ROOT=/usr/local/cuda

打开一个新的终端窗口，新建个py文件，输入下面代码进行测试

import torch
import torchvision

print(torch.cuda.is_available())
print(torchvision.__version__)

结果如下：