I can't use it by GPU - Githubissues

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch

https://www.adrianbulat.com

BSD 3-Clause "New" or "Revised" License

6.88k stars 1.33k forks source link

I can't use it by GPU #336

Closed xeiaobain closed 1 year ago

xeiaobain commented 1 year ago

Why can't I use the gpu to run the program after changing "CPU" to "cuda"? An error AssertionError is reported: Torch not compiled with CUDA enabled. I can run other programs, and the cuda and torch versions correspond. My English is not very good, please don't mind

DeNgKee commented 1 year ago

because your cuda's version is not compatible with you pytorch