CUDA_HOME environment variable is not set & No CUDA runtime is found

Hello,

I installed the UBUNTU 16.04 and Anaconda with python 3.7, pytorch 1.5, and CUDA 10.1 on my own computer. However, when I implement “python setup.py develop,” the error message “OSError: CUDA_HOME environment variable is not set” popped out. I used the “export CUDA_HOME=/usr/local/cuda-10.1” to try to fix the problem. Then, I re-run “python setup.py develop.” This time, a new error message popped out “No CUDA runtime is found, using CUDA_HOME=/usr/local/cuda-10.1” with “IndexError: list index out of range.” Anyone have any idea on how to fix this problem?

Thanks,
Kevin

Hi @kevinminion0918 ,
Wanted to check if you have CUDA installed on custom location?
System-wide installation at exactly /usr/local/cuda on Linux platforms. Versioned installation paths (i.e. /usr/local/cuda-12.0) are intentionally ignored.
You can use CUDA_HOME to select specific versions.

Thanks