Home Explore Help
Sign In
AI1
/
Algorithm-Repo
2
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Branch: master
Branches Tags
Algorithm-Repo
/
SourceCode
/
MachineLearningUtils
/
VINNOFramework
/
YOLO
/
docker
/
Dockerfile-arm64

Dockerfile-arm64 2.6 KB
Permalink History Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 
  1. # Ultralytics YOLO 🚀, AGPL-3.0 license
    2. 

  2. # Builds ultralytics/ultralytics:latest-arm64 image on DockerHub https://hub.docker.com/r/ultralytics/ultralytics
    3. 

  3. # Image is aarch64-compatible for Apple M1, M2, M3, Raspberry Pi and other ARM architectures
    4. 

  4. 

  5. # Start FROM Ubuntu image https://hub.docker.com/_/ubuntu with "FROM arm64v8/ubuntu:22.04" (deprecated)
    6. 

  6. # Start FROM Debian image for arm64v8 https://hub.docker.com/r/arm64v8/debian (new)
    7. 

  7. FROM arm64v8/debian:bookworm-slim
    8. 

  8. 

  9. # Set environment variables
    10. 

  10. ENV APP_HOME /usr/src/ultralytics
    11. 

  11. 

  12. # Downloads to user config dir
    13. 

  13. ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/Arial.ttf \
    14. 

  14.     https://github.com/ultralytics/assets/releases/download/v0.0.0/Arial.Unicode.ttf \
    15. 

  15.     /root/.config/Ultralytics/
    16. 

  16. 

  17. # Install linux packages
    18. 

  18. # g++ required to build 'tflite_support' and 'lap' packages, libusb-1.0-0 required for 'tflite_support' package
    19. 

  19. # pkg-config and libhdf5-dev (not included) are needed to build 'h5py==3.11.0' aarch64 wheel required by 'tensorflow'
    20. 

  20. RUN apt update \
    21. 

  21.     && apt install --no-install-recommends -y python3-pip git zip unzip wget curl htop gcc libgl1 libglib2.0-0 libpython3-dev gnupg g++ libusb-1.0-0
    22. 

  22. 

  23. # Create working directory
    24. 

  24. WORKDIR $APP_HOME
    25. 

  25. 

  26. # Copy contents and assign permissions
    27. 

  27. COPY . $APP_HOME
    28. 

  28. RUN git remote set-url origin https://github.com/ultralytics/ultralytics.git
    29. 

  29. ADD https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt $APP_HOME
    30. 

  30. 

  31. # Remove python3.11/EXTERNALLY-MANAGED to avoid 'externally-managed-environment' issue, Debian 12 Bookworm error
    32. 

  32. RUN rm -rf /usr/lib/python3.11/EXTERNALLY-MANAGED
    33. 

  33. 

  34. # Install pip packages
    35. 

  35. # Install tensorstore from .whl because PyPI does not include aarch64 binaries
    36. 

  36. RUN python3 -m pip install --upgrade pip wheel
    37. 

  37. RUN pip install --no-cache-dir https://github.com/ultralytics/yolov5/releases/download/v1.0/tensorstore-0.1.59-cp311-cp311-linux_aarch64.whl -e ".[export]"
    38. 

  38. 

  39. # Creates a symbolic link to make 'python' point to 'python3'
    40. 

  40. RUN ln -sf /usr/bin/python3 /usr/bin/python
    41. 

  41. 

  42. 

  43. # Usage Examples -------------------------------------------------------------------------------------------------------
    44. 

  44. 

  45. # Build and Push
    46. 

  46. # t=ultralytics/ultralytics:latest-arm64 && sudo docker build --platform linux/arm64 -f docker/Dockerfile-arm64 -t $t . && sudo docker push $t
    47. 

  47. 

  48. # Run
    49. 

  49. # t=ultralytics/ultralytics:latest-arm64 && sudo docker run -it --ipc=host $t
    50. 

  50. 

  51. # Pull and Run
    52. 

  52. # t=ultralytics/ultralytics:latest-arm64 && sudo docker pull $t && sudo docker run -it --ipc=host $t
    53. 

  53. 

  54. # Pull and Run with local volume mounted
    55. 

  55. # t=ultralytics/ultralytics:latest-arm64 && sudo docker pull $t && sudo docker run -it --ipc=host -v "$(pwd)"/shared/datasets:/usr/src/datasets $t
    56.