Trouble installing CUDA for stylegan2

harskish / ganspace

Discovering Interpretable GAN Controls [NeurIPS 2020]

Apache License 2.0

1.78k stars 264 forks source link

Closed duongpatwit closed 3 years ago

duongpatwit commented 3 years ago

I am having an issue with setting up stylegan2 at step 5 python setup.py install

I have installed CUDA toolkit version 10.1

I thought nothing of it and skipped to step 6 python -c "import torch; import upfirdn2d_op; import fused; print('OK')"

Anyone knows how to fix it or if I did something wrong?

duongpatwit commented 3 years ago

I fixed the problem, I had an outdated driver.