CUDA_VERSION: 11.6
Traceback (most recent call last):
File "/home/interns/dionysos/imaginaire/imaginaire/third_party/correlation/setup.py", line 35, in
CUDAExtension('correlation_cuda', [
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1074, in CUDAExtension
library_dirs += library_paths(cuda=True)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1201, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 2407, 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.
CUDA_VERSION: 11.6 Traceback (most recent call last): File "/home/interns/dionysos/imaginaire/imaginaire/third_party/correlation/setup.py", line 35, in
CUDAExtension('correlation_cuda', [
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1074, in CUDAExtension
library_dirs += library_paths(cuda=True)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1201, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/interns/dionysos/imaginaire/.conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 2407, 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.