Error when running the code "python setup.py bdist_wheel"

[SijanNeupane49]

python setup.py bdist_wheel
running bdist_wheel
running build
running build_py
running build_ext
/home/shrijan_pf/SoftGroup/lib/spconv/build/lib.linux-x86_64-3.7
Release
-- Caffe2: CUDA detected: 11.1
-- Caffe2: CUDA nvcc is: /usr/local/cuda-11.1/bin/nvcc
-- Caffe2: CUDA toolkit directory: /usr/local/cuda-11.1
-- Caffe2: Header version is: 11.1
CMake Warning (dev) at /home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/cmake/data/share/cmake-3.22/Modules/FindPackageHandleStandardArgs.cmake:438 (message):
The package name passed to find_package_handle_standard_args (CUDNN) does
not match the name of the calling package (Caffe2). This can lead to
problems in calling code that expects find_package result variables
(e.g., _FOUND) to follow a certain pattern.
Call Stack (most recent call first):
/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:107 (find_package_handle_standard_args)
/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include)
/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:40 (find_package)
CMakeLists.txt:23 (find_package)
This warning is for project developers. Use -Wno-dev to suppress it.

-- Found cuDNN: v? (include: /usr/include, library: /usr/lib/x86_64-linux-gnu/libcudnn.so)
CMake Error at /home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:159 (message):
PyTorch requires cuDNN 7 and above.
Call Stack (most recent call first):
/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include)
/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:40 (find_package)
CMakeLists.txt:23 (find_package)

-- Configuring incomplete, errors occurred!
See also "/home/shrijan_pf/SoftGroup/lib/spconv/build/temp.linux-x86_64-3.7/CMakeFiles/CMakeOutput.log".
See also "/home/shrijan_pf/SoftGroup/lib/spconv/build/temp.linux-x86_64-3.7/CMakeFiles/CMakeError.log".
Traceback (most recent call last):
File "setup.py", line 86, in
zip_safe=False,
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/setuptools/init.py", line 153, in setup
return distutils.core.setup(**attrs)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/core.py", line 148, in setup
dist.run_commands()
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/dist.py", line 966, in run_commands
self.run_command(cmd)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/wheel/bdist_wheel.py", line 299, in run
self.run_command('build')
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/command/build.py", line 135, in run
self.run_command(cmd_name)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "setup.py", line 39, in run
self.build_extension(ext)
File "setup.py", line 69, in build_extension
subprocess.check_call(['cmake', ext.sourcedir] + cmake_args, cwd=self.build_temp, env=env)
File "/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/subprocess.py", line 363, in check_call
raise CalledProcessError(retcode, cmd)
subprocess.CalledProcessError: Command '['cmake', '/home/shrijan_pf/SoftGroup/lib/spconv', '-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=/home/shrijan_pf/SoftGroup/lib/spconv/build/lib.linux-x86_64-3.7/spconv', '-DCMAKE_PREFIX_PATH=/home/shrijan_pf/anaconda3/envs/softgroup/lib/python3.7/site-packages/torch', '-DPYBIND11_PYTHON_VERSION=3.7', '-DSPCONV_BuildTests=OFF', '-DCMAKE_CUDA_FLAGS="--expt-relaxed-constexpr"', '-DCMAKE_BUILD_TYPE=Release']' returned non-zero exit status 1.
(softgroup) shrijan_pf@BQ-DX1100-CT2:~/SoftGroup/lib/spconv$

[thangvubk]

it seems the error is from here PyTorch requires cuDNN 7 and above.. Did you have your CUDNN installed.