Проверьте версию CUDA NN
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2orcat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
Hilarious Hamerkop
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2orcat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2