serengil / deepface

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python
https://www.youtube.com/watch?v=WnUVYQP4h44&list=PLsS_1RYmYQQFdWqxQggXHynP1rqaYXv_E&index=1
MIT License
13.65k stars 2.14k forks source link

Can not use on A100 cuda117 #840

Closed crj1998 closed 1 year ago

crj1998 commented 1 year ago
pip show deepface
Name: deepface
Version: 0.0.79
Summary: A Lightweight Face Recognition and Facial Attribute Analysis Framework (Age, Gender, Emotion, Race) for Python
Home-page: https://github.com/serengil/deepface
Author: Sefik Ilkin Serengil
Author-email: serengil@gmail.com
License: UNKNOWN
Location: /opt/conda/lib/python3.10/site-packages
Requires: fire, Flask, gdown, gunicorn, keras, mtcnn, numpy, opencv-python, pandas, Pillow, retina-face, tensorflow, tqdm
Required-by: 
Node: 'model/bn_data/FusedBatchNormV3'
cuDNN launch failure : input shape ([1,1536,1024,3])
         [[{{node model/bn_data/FusedBatchNormV3}}]] [Op:__inference_function_8077]
LiuShihHung commented 2 weeks ago

Hi, Do you solve this problem?