search

elliottzheng / face-detection

Fast and reliable face detection with RetinaFace.PyTorch
MIT License
114 stars 24 forks source link

可以在单机多卡上测试吗 #5

Open AliceIee99 opened 2 years ago

AliceIee99 commented 2 years ago

在detector = RetinaFace(gpu_id=0)中，gpu_id可以是多个值吗，我试着让gpu_id=[0,1,2,3]但遇到错误

elliottzheng commented 2 years ago

不支持，对于多个GPU你可以创建多个detector，每个detector在不同的gpu上，然后再通过啥多进程多线程地进行调用