From 1d5d105c62d8c5617c0dda441fdc6338fce26b1d Mon Sep 17 00:00:00 2001 From: Lakshantha Dissanayake Date: Thu, 25 Jul 2024 15:28:08 -0700 Subject: [PATCH] Add compatible `tensorstore` versions for `aarch64` (#14697) Co-authored-by: Glenn Jocher --- docker/Dockerfile-arm64 | 3 +-- pyproject.toml | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile-arm64 b/docker/Dockerfile-arm64 index d9ec7529..e7284e99 100644 --- a/docker/Dockerfile-arm64 +++ b/docker/Dockerfile-arm64 @@ -32,9 +32,8 @@ ADD https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt $A RUN rm -rf /usr/lib/python3.11/EXTERNALLY-MANAGED # Install pip packages -# Install tensorstore from .whl because PyPI does not include aarch64 binaries RUN python3 -m pip install --upgrade pip wheel -RUN pip install --no-cache-dir https://github.com/ultralytics/assets/releases/download/v0.0.0/tensorstore-0.1.59-cp311-cp311-linux_aarch64.whl -e ".[export]" +RUN pip install --no-cache-dir -e ".[export]" # Creates a symbolic link to make 'python' point to 'python3' RUN ln -sf /usr/bin/python3 /usr/bin/python diff --git a/pyproject.toml b/pyproject.toml index 203de68d..2a0c9b3b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -101,6 +101,7 @@ export = [ "openvino>=2024.0.0", # OpenVINO export "tensorflow>=2.0.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 "tensorflowjs>=3.9.0", # TF.js export, automatically installs tensorflow + "tensorstore>=0.1.63; platform_machine == 'aarch64' and python_version >= '3.9'", # for TF Raspberry Pi exports "keras", # not installed automatically by tensorflow>=2.16 "flatbuffers>=23.5.26,<100; platform_machine == 'aarch64'", # update old 'flatbuffers' included inside tensorflow package "numpy==1.23.5; platform_machine == 'aarch64'", # fix error: `np.bool` was a deprecated alias for the builtin `bool` when using TensorRT models on NVIDIA Jetson