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/

相关推荐
Patrick_Wilson4 天前
从「改个端口」到 502:Next.js on k8s 的容器端口、Service 映射与 env 覆盖
docker·kubernetes·next.js
Suroy5 天前
DockerView-Go:用 Go 写一个终端 Docker 监控工具,顺便做了个 Web 仪表盘
docker
云恒要逆袭5 天前
运行你的第一个Docker容器
后端·docker·容器
宋均浩6 天前
# Docker 镜像瘦身实战:从 1.2G 到 80MB 的五个优化步骤
ci/cd·docker
程序员老赵6 天前
10 分钟部署 OpenCode:Docker 一键安装,浏览器打开就能用 AI 写代码(附完整命令与排错)
docker·容器·ai编程
WangMingHua1117 天前
LM Studio Docker 部署——本地大模型一键启动
docker
曲幽8 天前
别再用网页翻译看源码了!你的私人翻译神器LibreTranslate,部署避坑指南来了
python·docker·web·pot·translate·libretranslate·arogstranslate
武子康10 天前
调查研究-183 Apple container:Mac 上用轻量 VM 跑 Linux 容器,Swift 会改写本地容器体验吗?
docker·容器·apple
Alsn8613 天前
等待学习-学习目录:Docker 容器安全攻防
学习·安全·docker
程序员老赵13 天前
服务器没有桌面?Docker 跑个 Chrome,浏览器就能远程用
docker·容器·devops