Can TE optimize the find cudnn?
MARD1NO opened this issue · 1 comments
MARD1NO commented
Maybe user do not install cuda/cudnn toolkit independently, but pytorch will pack their own cudnn in installation package.
I think TE can set the default cudnn path by pytorch install dir, otherwise user need to specify the cudnn include directory by setting ENV_VAR CUDNN_INCLUDE_DIR.