Tencent / DVQA

Deep learning-based Video Quality Assessment
470 stars 89 forks source link

How to use it by CPU? #9

Closed liu0527aa closed 4 years ago

liu0527aa commented 4 years ago

I have prepared the envirionment.But I want to use by CPU.Could you tell me or add the way about how to use it by CPU instead of GPU

tommyhq commented 4 years ago

We recommend the following way to disable GPU. Modify device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") to device = torch.device("cpu") and remove multiple GPU option in your command line --multi_gpu

I believe you know it is much slower to run it on CPU.