diff --git a/docker/Dockerfile-jetson-jetpack6 b/docker/Dockerfile-jetson-jetpack6 index d6d61176..9cd6afdb 100644 --- a/docker/Dockerfile-jetson-jetpack6 +++ b/docker/Dockerfile-jetson-jetpack6 @@ -1,9 +1,9 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license # Builds ultralytics/ultralytics:jetson-jetpack6 image on DockerHub https://hub.docker.com/r/ultralytics/ultralytics -# Supports JetPack6.x for YOLO11 on Jetson AGX Orin, Orin NX and Orin Nano Series +# Supports JetPack6.1 for YOLO11 on Jetson AGX Orin, Orin NX and Orin Nano Series # Start FROM https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-jetpack -FROM nvcr.io/nvidia/l4t-jetpack:r36.3.0 +FROM nvcr.io/nvidia/l4t-jetpack:r36.4.0 # Set environment variables ENV PYTHONUNBUFFERED=1 \ @@ -17,9 +17,11 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/Arial.ttf \ /root/.config/Ultralytics/ # Install dependencies -RUN apt-get update && \ +ADD https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/arm64/cuda-keyring_1.1-1_all.deb . +RUN dpkg -i cuda-keyring_1.1-1_all.deb && \ + apt-get update && \ apt-get install -y --no-install-recommends \ - git python3-pip libopenmpi-dev libopenblas-base libomp-dev \ + git python3-pip libopenmpi-dev libopenblas-base libomp-dev libcusparselt0 libcusparselt-dev \ && rm -rf /var/lib/apt/lists/* # Create working directory @@ -36,9 +38,9 @@ ADD https://nvidia.box.com/shared/static/48dtuob7meiw6ebgfsfqakc9vse62sg4.whl on # Pip install onnxruntime-gpu, torch, torchvision and ultralytics RUN python3 -m pip install --upgrade pip uv RUN uv pip install --system \ - onnxruntime_gpu-1.18.0-cp310-cp310-linux_aarch64.whl \ - https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.3.0-cp310-cp310-linux_aarch64.whl \ - https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl + https://github.com/ultralytics/assets/releases/download/v0.0.0/onnxruntime_gpu-1.20.0-cp310-cp310-linux_aarch64.whl \ + https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.5.0a0+872d972e41.nv24.08-cp310-cp310-linux_aarch64.whl \ + https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.20.0a0+afc54f7-cp310-cp310-linux_aarch64.whl RUN uv pip install --system -e ".[export]" # Remove extra build files