Pip install spatial_correlation_sampler

error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [60 lines of output]
No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda’
running bdist_wheel
running build
running build_py
creating build
creating build/lib.linux-x86_64-cpython-38
creating build/lib.linux-x86_64-cpython-38/spatial_correlation_sampler
copying Correlation_Module/spatial_correlation_sampler/spatial_correlation_sampler.py → build/lib.linux-x86_64-cpython-38/spatial_correlation_sampler
copying Correlation_Module/spatial_correlation_sampler/init.py → build/lib.linux-x86_64-cpython-38/spatial_correlation_sampler
running build_ext
Traceback (most recent call last):
File “”, line 2, in
File “”, line 34, in
File “/tmp/pip-install-wja02f1l/spatial-correlation-sampler_76cbda975a75419c9cfb3bb63fdec5a9/setup.py”, line 57, in
launch_setup()
File “/tmp/pip-install-wja02f1l/spatial-correlation-sampler_76cbda975a75419c9cfb3bb63fdec5a9/setup.py”, line 25, in launch_setup
setup(
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/init.py”, line 107, in setup
return distutils.core.setup(**attrs)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/core.py”, line 185, in setup
return run_commands(dist)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/core.py”, line 201, in run_commands
dist.run_commands()
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/dist.py”, line 969, in run_commands
self.run_command(cmd)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/dist.py”, line 1234, in run_command
super().run_command(command)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/dist.py”, line 988, in run_command
cmd_obj.run()
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/wheel/bdist_wheel.py”, line 325, in run
self.run_command(“build”)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/cmd.py”, line 318, in run_command
self.distribution.run_command(command)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/dist.py”, line 1234, in run_command
super().run_command(command)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/dist.py”, line 988, in run_command
cmd_obj.run()
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/command/build.py”, line 131, in run
self.run_command(cmd_name)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/cmd.py”, line 318, in run_command
self.distribution.run_command(command)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/dist.py”, line 1234, in run_command
super().run_command(command)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/dist.py”, line 988, in run_command
cmd_obj.run()
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/command/build_ext.py”, line 84, in run
_build_ext.run(self)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/Cython/Distutils/old_build_ext.py”, line 186, in run
_build_ext.build_ext.run(self)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py”, line 345, in run
self.build_extensions()
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/torch/utils/cpp_extension.py”, line 499, in build_extensions
_check_cuda_version(compiler_name, compiler_version)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/torch/utils/cpp_extension.py”, line 383, in _check_cuda_version
torch_cuda_version = packaging.version.parse(torch.version.cuda)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/pkg_resources/_vendor/packaging/version.py”, line 52, in parse
return Version(version)
File “/home/unbuntu/anaconda3/envs/fubao/lib/python3.8/site-packages/pkg_resources/_vendor/packaging/version.py”, line 196, in init
match = self._regex.search(version)
TypeError: expected string or bytes-like object
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for spatial_correlation_sampler
Running setup.py clean for spatial_correlation_sampler
Failed to build spatial_correlation_sampler
ERROR: Could not build wheels for spatial_correlation_sampler, which is required to install pyproject.toml-based projects