Thanks for your reply. However it does not work, and blow is buidling log
running build_ext
-- Building with NumPy bindings
-- Not using cuDNN
-- Detected CUDA at /usr/local/cuda
-- Building NCCL library
building 'torch._C' extension

Is there any other variable needed to specify CUDNN? I just build the pytorch: python setup.py build