运行python -u infer_audio2vid.py报错

[BoragoCode]

运行环境： python3.8 torch版本： torch 2.2.2 torchaudio 2.2.2 torchvision 0.17.2 ubuntu系统： No LSB modules are available. Distributor ID: Ubuntu Description: Ubuntu 22.04.1 LTS Release: 22.04 Codename: jammy 运行命令： python -u infer_audio2vid.py 报错： File "infer_audio2vid.py", line 257, in main() File "infer_audio2vid.py", line 200, in main det_bboxes, probs = face_detector.detect(face_img) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/facenet_pytorch/models/mtcnn.py", line 313, in detect batch_boxes, batch_points = detect_face( File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/facenet_pytorch/models/utils/detect_face.py", line 73, in detect_face reg, probs = pnet(im_data) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(*args, *kwargs) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/facenet_pytorch/models/mtcnn.py", line 38, in forward x = self.conv1(x) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(args, kwargs) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 460, in forward return self._conv_forward(input, self.weight, self.bias) File "/home/sczc/anaconda3/envs/py3.8_echo_mimic/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 456, in _conv_forward return F.conv2d(input, weight, bias, self.stride, RuntimeError: GET was unable to find an engine to execute this computation

[lymhust]

mtcnn人脸检测的环境没配置对，检查一下这个模块。看看是环境问题，还是模型加载问题。

[11Boss]

我遇到了跟你同样的问题，是不是cuda的驱动版本太低了不支持torch2.0

[nitinmukesh]

@BoragoCode @11Boss

Please see if this helps https://www.youtube.com/watch?v=WtHdvSSQlWo