Skip to content

python setup.py bdist_wheel出现错误 #42

Open
@WS517

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions