Cannot run in cpu only mode via docker

[risharde]

Hi, I'm getting the following error trying to run the standard docker (except I removed --gpus 'all' part since that causes an absolutely fail in execution) command for SadTalker I do not have an nvidia gpu and would like to run it on the cpu. Any ideas if this can be fixed (or support for AMD GPUs?)

D:\ai_docker\sadtalker>docker run --rm -v D:\ai_docker\sadtalker:/host_dir wawa9000/sadtalker --driven_audio /host_dir/audio.wav --source_image /host_dir/photo.png --expression_scale 1.0 --cpu --still --result_dir /host_dir ./checkpoints/epoch_20.pth Traceback (most recent call last): File "inference.py", line 131, in main(args) File "inference.py", line 48, in main preprocess_model = CropAndExtract(path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting, device) File "/app/SadTalker/src/utils/preprocess.py", line 46, in init self.kp_extractor = KeypointExtractor() File "/app/SadTalker/src/face3d/extract_kp_videos.py", line 16, in init self.detector = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D) File "/usr/local/lib/python3.8/site-packages/face_alignment/api.py", line 77, in init self.face_detector = face_detector_module.FaceDetector(device=device, verbose=verbose, **face_detector_kwargs) File "/usr/local/lib/python3.8/site-packages/face_alignment/detection/sfd/sfd_detector.py", line 31, in init self.face_detector.to(device) File "/usr/local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 927, in to return self._apply(convert) File "/usr/local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 579, in _apply module._apply(fn) File "/usr/local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 602, in _apply param_applied = fn(param) File "/usr/local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 925, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) File "/usr/local/lib/python3.8/site-packages/torch/cuda/init.py", line 217, in _lazy_init torch._C._cuda_init() RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

[risharde]

@vinthony tagged

[Maximus48p]

same issue..