search

pkhungurn / talking-head-anime-2-demo

Demo programs for the Talking Head Anime from a Single Image 2: More Expressive project.
http://pkhungurn.github.io/talking-head-anime-2/
MIT License
1.12k stars 154 forks source link

Install pytorch in conda (Windows) #32

Closed itemx closed 2 years ago

itemx commented 2 years ago

As the time goes by, current sample command in readme will not install a proper pytorch with cuda support. According to pytorch's official site, the latest windows version does not support of CUDA 10.2 anymore.

Insted I tried install 11.3 as recommend (And also required for cards like 3060)

conda install pytorch torchvision cudatoolkit=11.3 -c pytorch

Then it works. I don't have git client in my recent environment so is unable to push, but I think this information can be shared in readme as well?

dragonmeteor commented 2 years ago

Sorry for the long silence. I've updated the README file.