Jetson orin agx配置ultralytics 使用docker或conda

一、docker

以yolov8为例进行探索:

https://www.jetson-ai-lab.com/tutorial_ultralytics#__tabbed_2_1

  1. 拉取对应docker镜像

注意要先把原先的东西删除彻底了再安装

停止服务

sudo systemctl stop docker.service

安装后赶紧先改默认路径位置 + 换源

sudo vim /etc/docker/daemon.json

docker Root Dir: /home/user/docker_work/docker

重启

sudo systemctl daemon-reexec

sudo systemctl start docker

  1. 进入镜像

启动容器:

t=ultralytics/ultralytics:latest-jetson-jetpack6

sudo docker pull t \&\& sudo docker run -it --ipc=host --runtime=nvidia --network=host t

配置http代理:

root@ubuntu:/ultralytics# export http_proxy="http://127.0.0.1:7897"

root@ubuntu:/ultralytics# export https_proxy="http:/127.0.0.1:7897"

配置pip源

设置默认镜像源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/

设置信任主机

pip config set global.trusted-host pypi.tuna.tsinghua.edu.cn

二、conda配置

如果不想用docker 怎么配置呢:以为ultralytics为例

首先要确认自己的版本信息

  1. Cuda版本 nvcc 这里是12.6

  2. 安装好jet-pack, 确认对应版本, 这里是6.2.1

https://pypi.jetson-ai-lab.io/jp6/cu126 这里便会有jetson的一些python轮子,可以看出都要求python=3.10

  1. 创建虚拟环境 注意创建前最好更新一下conda, 注意指定python版本

  2. 根据上述网址,下载torch torchvision onnx 的whl并安装

  3. Pip install ultralytics numpy==1.26.1 torch==2.8.0总而言之,一定要指定你必须要使用的版本,否则pip会直接给你下载兼容ultralytics的最新的torch, 这个torch可就未必兼容你的jetpack了

  4. 还没有装tensorrt, 这个网址里是找不到的,但是jetpack已经给你装过了,在/usr/lib/python3.10/dist-packages, 软链接过去,注意换成自己的路径。

设置一下路径变量,注意先进入你要配置的conda虚拟环境

CONDA_SITE_PACKAGES=$(python -c 'import site; print(site.getsitepackages()[-1])')

SYSTEM_PATH="/usr/lib/python3.10/dist-packages"

链接核心模块文件夹

sudo ln -s SYSTEM_PATH/tensorrt CONDA_SITE_PACKAGES/tensorrt

sudo ln -s SYSTEM_PATH/tensorrt_dispatch CONDA_SITE_PACKAGES/tensorrt_dispatch

sudo ln -s SYSTEM_PATH/tensorrt_lean CONDA_SITE_PACKAGES/tensorrt_lean

复制元数据(让 pip 和 Conda 识别版本信息)

sudo cp -r SYSTEM_PATH/tensorrt-10.3.0.dist-info CONDA_SITE_PACKAGES/

sudo cp -r SYSTEM_PATH/tensorrt_dispatch-10.3.0.dist-info CONDA_SITE_PACKAGES/

sudo cp -r SYSTEM_PATH/tensorrt_lean-10.3.0.dist-info CONDA_SITE_PACKAGES/

相关推荐
方二华3 小时前
5 mysql源码中B+树的构建
数据库·mysql·1024程序员节
星火科技探索实验室3 小时前
【实战经验】飞牛云 如何使用 SSD 缓存加速?
1024程序员节
千百元3 小时前
java 程序Apache log4j JDBCAppender SQL注入漏洞(CVE-2022-23305)
1024程序员节
开心-开心急了3 小时前
Kivy 乒乓游戏教程 基于Minconda或Anconda 运行
python·conda·1024程序员节·kivy
小彭律师5 小时前
Docker/K8s部署MySQL的创新实践与优化技巧大纲
mysql·docker·kubernetes
lastHertz5 小时前
Docker 占用导致 C 盘空间不足的排查与解决
运维·docker·容器
专家大圣5 小时前
Docker+Redis监控新方案:cpolar让远程管理“零配置”
网络·redis·docker·容器·内网穿透
诗句藏于尽头6 小时前
自动签到之实现掘金模拟签到
python·1024程序员节
chen_note9 小时前
Dockerfile及其部署镜像步骤
docker·容器·镜像·dockerfile