浏览代码

lock libedgetpu1 to 15.0, update tflite_runtime

Michael Wei 4 年之前
父节点
当前提交
a85d780020
共有 4 个文件被更改,包括 6 次插入6 次删除
  1. 1 1
      docker/Dockerfile.aarch64
  2. 3 3
      docker/Dockerfile.amd64
  3. 1 1
      docker/Dockerfile.armv7
  4. 1 1
      docker/Dockerfile.base

+ 1 - 1
docker/Dockerfile.aarch64

@@ -17,6 +17,6 @@ RUN apt-get -qq update \
         libtiff5 \
         libdc1394-22 \
     ## Tensorflow lite
-    && pip3 install https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp38-cp38-linux_aarch64.whl \
+    && pip3 install https://github.com/google-coral/pycoral/releases/download/release-frogfish/tflite_runtime-2.5.0-cp38-cp38-linux_aarch64.whl \
     && rm -rf /var/lib/apt/lists/* \
     && (apt-get autoremove -y; apt-get autoclean -y)

+ 3 - 3
docker/Dockerfile.amd64

@@ -11,8 +11,8 @@ RUN apt-get -qq update \
         # VAAPI drivers for Intel hardware accel
         libva-drm2 libva2 i965-va-driver vainfo intel-media-va-driver \
         ## Tensorflow lite
-    && wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
-    && python3.8 -m pip install tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
-    && rm tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
+    && wget -q https://github.com/google-coral/pycoral/releases/download/release-frogfish/tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
+    && python3.8 -m pip install tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
+    && rm tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
     && rm -rf /var/lib/apt/lists/* \
     && (apt-get autoremove -y; apt-get autoclean -y)

+ 1 - 1
docker/Dockerfile.armv7

@@ -19,6 +19,6 @@ RUN apt-get -qq update \
         libaom0 \
         libx265-179 \
     ## Tensorflow lite
-    && pip3 install https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp38-cp38-linux_armv7l.whl \
+    && pip3 install https://github.com/google-coral/pycoral/releases/download/release-frogfish/tflite_runtime-2.5.0-cp38-cp38-linux_armv7l.whl \
     && rm -rf /var/lib/apt/lists/* \
     && (apt-get autoremove -y; apt-get autoclean -y)

+ 1 - 1
docker/Dockerfile.base

@@ -23,7 +23,7 @@ RUN apt-get -qq update \
     && echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" > /etc/apt/sources.list.d/coral-edgetpu.list \
     && echo "libedgetpu1-max libedgetpu/accepted-eula select true" | debconf-set-selections \
     && apt-get -qq update && apt-get -qq install --no-install-recommends -y \
-        libedgetpu1-max \
+        libedgetpu1-max=15.0 \
     && rm -rf /var/lib/apt/lists/* /wheels \
     && (apt-get autoremove -y; apt-get autoclean -y)