import SSD._C as C ImportError: /usr/local/lib/python3.6/dist-packages/SSD/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: cudnnFindConvolutionForwardAlgorithmEx

[HangJie720]

I try to compile apex from source based on official pytorch1.8.1 whl: torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl,

python setup.py build --cpp_ext --cuda_ext
python setup.py bdist_wheel --cpp_ext --cuda_ext -d build_pip
pip3 install build_pip/apex-0.1-cp36-cp36m-linux_x86_64.whl

root@5e359b5d8031:training_results_v0.7/NVIDIA/benchmarks/ssd/implementations/pytorch# python -u -m bind_launch --nproc_per_node 2 --nsockets_per_node 2 --ncores_per_socket 8 train.py --no-dali --dali-cache 0 --delay-allreduce --batch-size 16 --data=/home/datasets/coco/

import SSD._C as C
ImportError: /usr/local/lib/python3.6/dist-packages/SSD/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: cudnnFindConvolutionForwardAlgorithmEx