pkhungurn / talking-head-anime-demo

Demo for the "Talking Head Anime from a Single Image."
MIT License
1.99k stars 286 forks source link

How to replace the Cuda driver in the AMD Product. #27

Closed Kazto-lsk closed 3 years ago

Kazto-lsk commented 3 years ago

Always stay healthy.

I think this project is very positive, But the graphics card I have now is RX580. Is there any way the AMD product line can do in this project? If not, what is the best product to use this project?

dragonmeteor commented 3 years ago

The short answer is no. It might be possible by porting the network to a more universal format with some other tools, but you'd have to do it by yourself.

The code is best run on a graphics card that supports CUDA. I've personally tested the network on the GeForce GTX 1080 Ti and the Titan RTX.