search

XingangPan / DragGAN

Official Code for DragGAN (SIGGRAPH 2023)
https://vcai.mpi-inf.mpg.de/projects/DragGAN/
Other
35.65k stars 3.45k forks source link

Cudatoolkit issues #128

Closed omarirfa closed 1 year ago

omarirfa commented 1 year ago

Has anyone faced the issue for cudatoolkit on windows using conda? I get this error:

image

Do I need to download cudatoolkit from Nvidia manually?

PDillis commented 1 year ago

Set the CUDA_HOME variable in the PATH. In Windows you just look for Path in the serach bar and you'll get to System Properties, click on Environment Variables and you can add a New System variable as CUDA_HOME and set it to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8 (this is my specific case, you should see where you have CUDA installed).

In Ubuntu it's a bit easier, as you simply set it like so in the command line: export CUDA_HOME=/usr/local/cuda-X.X (with whichever version of CUDA you have, assuming that's where it's installed).

omarirfa commented 1 year ago

Thanks worked like a charm!