Pytorch-Correlation-extension
Pytorch-Correlation-extension copied to clipboard
Facing issue installing spatial correlation sampler
I am trying to install spatial correlation sampler packager for one of my application. I am installing it through pip install spatial-correlation-sampler
When ever I try to install I am getting an error
Traceback (most recent call last): File "C:\Program Files\Python311\Lib\site-packages\pip_vendor\pep517\in_process_in_process.py", line 351, in
main() File "C:\Program Files\Python311\Lib\site-packages\pip_vendor\pep517\in_process_in_process.py", line 333, in main json_out['return_val'] = hook(**hook_input['kwargs']) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Program Files\Python311\Lib\site-packages\pip_vendor\pep517\in_process_in_process.py", line 152, in prepare_metadata_for_build_wheel return hook(metadata_directory, config_settings) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Program Files\Python311\Lib\site-packages\setuptools\build_meta.py", line 377, in prepare_metadata_for_build_wheel self.run_setup() File "C:\Program Files\Python311\Lib\site-packages\setuptools\build_meta.py", line 484, in run_setup self).run_setup(setup_script=setup_script) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Program Files\Python311\Lib\site-packages\setuptools\build_meta.py", line 335, in run_setup exec(code, locals()) File " ", line 69, in File " ", line 48, in launch_setup File "C:\Program Files\Python311\Lib\site-packages\torch\utils\cpp_extension.py", line 1076, in CUDAExtension library_dirs += library_paths(cuda=True) ^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Program Files\Python311\Lib\site-packages\torch\utils\cpp_extension.py", line 1214, in library_paths paths.append(_join_cuda_home(lib_dir)) ^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Program Files\Python311\Lib\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.
My echo output
echo %CUDA_HOME% C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6
Even though cuda tool kit is installed and this CUDA_HOME env is set I am getting the same error. I am trying this in windows os. If there is any solution kindly let me know