CUDA_HOME environment variable is not set. Please set it to your CUDA install root

DDP/GPU
1

Hi everyone,

I’ve encountered the following error:

Traceback (most recent call last):
  File "/teamspace/studios/this_studio/Pointcept/libs/pointops/setup.py", line 26, in <module>
    CUDAExtension(
  File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1076, in CUDAExtension
    library_dirs += library_paths(cuda=True)
  File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1207, in library_paths
    if (not os.path.exists(_join_cuda_home(lib_dir)) and
  File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2416, in _join_cuda_home
    raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

Does anyone have any suggestions on how to resolve this? I’ve already added the following lines to my .bashrc file, but it hasn’t helped:

export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

Any insights on this issue would be greatly appreciated.