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