I have new GPU driver,but.....

[albbzmkh]

Exception in Tkinter callback
Traceback (most recent call last):
File "D:\PYTHON\anaconda\lib\tkinter_init_.py", line 1705, in call
return self.func(*args)
File "D:/OneDrive/Python/aa/talking-head-anime-demo/app/puppeteer.py", line 94, in load_image
self.load_image_from_file(file_name)
File "D:/OneDrive/Python/aa/talking-head-anime-demo/app/puppeteer.py", line 106, in load_image_from_file
self.source_image = extract_pytorch_image_from_filelike(file_name).to(self.torch_device).unsqueeze(dim=0)
File "D:\PYTHON\anaconda\lib\site-packages\torch\cuda_init_.py", line 196, in _lazy_init
check_driver()
File "D:\PYTHON\anaconda\lib\site-packages\torch\cuda_init.py", line 101, in _check_driver
http://www.nvidia.com/Download/index.aspx""")
AssertionError:
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
[ WARN:0] terminating async callback

[dragonmeteor]

Sorry for the late reply.

First, let us confirm one thing. Do you have an Nvidia GPU? You said you have a new driver, but you didn't say you have such a GPU.

[sheiun]

@albbzmkh You need to make sure torch.cuda.device_count() is greater than 0

[drojf]

@dragonmeteor you may want to explicitly add to the readme that you need a NVIDIA GPU/Cuda capable gpu to use this, as it's not stated there as a requirement.

I did try running in CPU-only mode, but since I don't really know what I'm doing, I'm unsure whether it's CPU-only is supported or if I was doing something else wrong. I assume it's just not supported.

edit: After changing the appropriate line to use CPU mode, it's definitely too slow to use in CPU mode, a GPU is required.

Exception in Tkinter callback
Traceback (most recent call last):
  File "C:\Users\drojf\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1705, in __call__
    return self.func(*args)
  File "C:\Users\drojf\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 749, in callit
    func(*args)
  File "app/manual_poser.py", line 102, in update_image
    self.update_pose()
  File "app/manual_poser.py", line 96, in update_pose
    self.current_pose = torch.zeros(self.pose_size, device=self.torch_device)
  File "C:\Users\drojf\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\cuda\__init__.py", line 196, in _lazy_init
    _check_driver()
  File "C:\Users\drojf\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\cuda\__init__.py", line 94, in _check_driver
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

[dragonmeteor]

I've updated the README to say that a GPU is required.

You can definitely run the code with only a CPU, but it is hopelessly slow.