运行环境:
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
运行环境: 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