오늘의 인기 글
최근 글
최근 댓글
Today
Total
05-03 06:06
관리 메뉴

우노

[Coral TPU] 환경 설치 본문

Hardware/Coral

[Coral TPU] 환경 설치

운호(Noah) 2022. 7. 5. 14:35

들어가기 앞서,

  • 해당 포스팅에선, edge tpu 내부 컨테이너에서, 이미지 추론을 진행하기 위해 설치해야했던 라이브러리들의 설치 코드를 정리합니다.
    1. libedgetpu1-std (v15.0)
    2. python3-edgetpu(v15.0)
    3. tflite-runtime (v2.1.0)
  • 설치 이후, 반드시 컨테이너를 재부팅해야 라이브러리들이 정상적으로 적용됩니다.
  • 그럼에도 불구하고 적용이 안된다면,
  • Coral USB를 다시 연결하거나, 호스트(해당 실험에선 라즈베리파이)를 재부팅해야합니다.

라이브러리 설치 코드

# Install Edge TPU Libraries
echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" | tee /etc/apt/sources.list.d/coral-edgetpu.list
curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | apt-key add -
apt-get update

# Libraries required for Mini PCIe Coral Chip
apt-get install -y gasket-dkms libedgetpu1-std usbutils 
apt-get install -y python3-edgetpu

# tflite runtime install(arm 64, python 3.6)
pip3 install https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp36-cp36m-linux_aarch64.whl

# package, library check
dpkg -l | grep edgetpu
# ii  libedgetpu1-legacy-std:arm64         15.0     arm64        Support library for Edge TPU
# ii  python3-edgetpu                      15.0     arm64        Edge TPU Python API

pip3 list | grep tflite-runtime
# tflite-runtime          2.1.0.post1

# device check
lsusb
# Bus 002 Device 001: ID 1d6b:0003 Linux Foundation 3.0 root hub
# Bus 001 Device 005: ID 18d1:9302 Google Inc. 
# Bus 001 Device 002: ID 2109:3431 VIA Labs, Inc. Hub
# Bus 001 Device 001: ID 1d6b:0002 Linux Foundation 2.0 root hub

############################## 컨테이너 재부팅 필요 ##############################

라이브러리 설치 전 발생했던 에러

ValueError: Failed to load delegate from libedgetpu.so.1
RuntimeError: Internal: Unsupported data type in custom op handler: 0Node number 1 (EdgeTpuDelegateForCustomOp) failed to prepare.
Comments