TensorFlow 2.10 & YOLO & MediaPipe & DeepFace 설치

처음에 기본적으로 TensorFlow를 깔 때는 아래 블로그를 따라하였습니다.

Anaconda에서 GPU 사용 가능한 Tensorflow 설치하기

 

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

 

이런식으로 뜨면 성공입니다! 

 

윈도우 환경에서 TensorFlow를 사용하기 위해서는 2.10버전을 사용하였습니다.

하지만 프로젝트를 진행하면서 MediaPipe와 Yolo, DeepFace를 추가 설치하였는데, 

Descriptors cannot be created directly

AttributeError: _ARRAY_API not found

module 'mediapipe' has no attribute 'solutions'

 

위와 같은 에러들이 떴습니다. 

Protobuf, Numpy, Mediapipe 에러인데,

 

먼저 Protobuf 에러는 코드 위에 아래와 같은 환경변수를 강제로 해준 뒤 처리해주었습니다.

import os
os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"

 

그리고 NumPy 버전은 1.x 버전으로 맞춰주었습니다. 2.x는 TensorFlow가 인식을 못했습니다.

하지만 MediaPipe를 깔면서 계속 NumPy가 2.x버전으로 업그레이드 되어서 다시 변경해주었습니다. 

pip uninstall protobuf numpy mediapipe tensorflow tensorflow-intel -y

pip install tensorflow==2.10.0 "numpy<2.0" "protobuf==3.20.1" "mediapipe<=0.10.11"

 

위의 코드를 순차적으로 해주면 됩니다! 마지막으로 나머지 라이브러리를 추가합니다.

pip install ultralytics deepface tensorrt-cu11

 

만약 아래 코드가 에러 없이 돌아가면 성공입니다!

import os
os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"

import numpy as np
import tensorflow as tf
import mediapipe as mp
from deepface import DeepFace

print(f"NumPy 버전: {np.__version__}") # 1.x.x 여야 함
print(f"TF GPU 가속: {tf.config.list_physical_devices('GPU')}")