Open
Description
我使用的是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)
请问这种情况应该如何解决,谢谢
Metadata
Assignees
Labels
No labels