Skip to content

Commit

Permalink
Fix tfjs exports for ARM64 Dockerfile (ultralytics#8766)
Browse files Browse the repository at this point in the history
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
  • Loading branch information
2 people authored and hmurari committed Apr 17, 2024
1 parent ce9b1e0 commit 9dc2388
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion docker/Dockerfile-arm64
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,12 @@ RUN rm -rf /usr/lib/python3.11/EXTERNALLY-MANAGED

# Install pip packages
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache -e .
RUN pip install --no-cache -e ".[export]"

# Fix error: `np.object` was a deprecated alias for the builtin `object` when exporting to TensorFlow.js
# Here onnxruntime is installed first because in tflite and onnx simplify exports, it will install latest numpy version as a dependency and we need to downgrade it
RUN pip install --no-cache onnxruntime
RUN pip install --no-cache numpy==1.23.5

# Creates a symbolic link to make 'python' point to 'python3'
RUN ln -sf /usr/bin/python3 /usr/bin/python
Expand Down

0 comments on commit 9dc2388

Please sign in to comment.