search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
7.05k stars 1.34k forks source link

TypeError: __init__() got an unexpected keyword argument 'enable_cuda' #113

Closed shliang0603 closed 5 years ago

shliang0603 commented 5 years ago

@1adrianb My environment is Pytorch 0.4.0 and I installed enable cuda env. image

When I modified the parameter，the error is following image image Can you give me advice ,thanks

1adrianb commented 5 years ago

Hi @shliang0603 as you can see in the last error message you attempt o use 'cuda' without compiling pytorch with cuda support. You can either do that ore use device='cpu' and run it on the cpu, but this will be much slower.