CUDA_HOME environment variable is not set about extension-cpp HOT 8 CLOSED

That's odd. Maybe you have an unusual install location for CUDA. Either way, just setting CUDA_HOME to your cuda install path before running python setup.py should work: CUDA_HOME=/path/to/your/cuda/home python setup.py install

from extension-cpp.

@zzd1992 Could you tell how to solve the problem about "the CUDA_HOME environment variable is not set"?

from extension-cpp.

The problem could be solved by installing the whole cuda through the nvida website. and when installing it, you may come across some problem. then https://askubuntu.com/questions/1280205/problem-while-installing-cuda-toolkit-in-ubuntu-18-04/1315116#1315116?newreg=ec85792ef03b446297a665e21fff5735 the answer may be to help you. GOOD LUCK

from extension-cpp.

Thanks. My problem is solved.

from extension-cpp.

@DaLei001

CUDA_HOME=/path/to/your/cuda/home python setup.py install

This is worked in my situation.

from extension-cpp.

I am getting this error in a conda env on a server and I have cudatoolkit installed on the conda env. What should the CUDA_HOME be in my case.

from extension-cpp.

@mmahdavian cudatoolkit probably won't work for you, it doesn't provide access to low level c++ apis. You'd need to install CUDA using the official method.

from extension-cpp.

I got a similar error when using pycharm, with unusual cuda install location.

I just add the CUDA_HOME env and solve this problem.

from extension-cpp.