python setup.py bdist_wheel出现错误

[SakuraTau]

我使用的是ubuntu20.04+cuda11.3+cudnn8.2.1+torch1.12.1+spconv-cuda11.3的版本是2.3.6，cmake版本为3.28.0，在运行1 python setup.py bdist_wheel时候报错 Found cuDNN: v? (include: /usr/local/cuda-11.3/include, library: /usr/local/cuda-11.3/lib64/libcudnn.so) CMake Error at /home/ws/miniconda3/envs/hais/lib/python3.7/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:185 (message): PyTorch requires cuDNN 7 and above. Call Stack (most recent call first): /home/ws/miniconda3/envs/hais/lib/python3.7/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include) /home/ws/miniconda3/envs/hais/lib/python3.7/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:68 (find_package) CMakeLists.txt:22 (find_package) 请问这种情况应该如何解决，谢谢

[Greatest-Marx]

检查pycorch的版本与cuda、cudnn是否匹配，如果匹配的话检查cmake版本，版本太高也不太行（建议cmake==3.14.4）。我感觉配置环境就是玄学，不同的电脑会遇到不同的问题，我把配置HAIS环境的问题总结了一下，希望能对你有帮助。总结发布在csdn：https://blog.csdn.net/TravleGun/article/details/130875949?spm=1001.2014.3001.5502