jiupinjia / SkyAR

Official Pytorch implementation of the preprint paper "Castle in the Sky: Dynamic Sky Replacement and Harmonization in Videos", in arXiv:2010.11800.
https://jiupinjia.github.io/skyar/
2.01k stars 248 forks source link

请问一下NVIDIA显卡是不是必要条件呢? #10

Closed wolfaherd closed 3 years ago

jiupinjia commented 3 years ago

Hi, Thanks for your question. Yes, NVIDIA cards are necessary for PyTorch GPU programming. You cannot use an AMD card in PyTorch (Tensorflow also does not support AMD btw). However, you may still run this project in CPU mode if you want, although it would be much slower.