Проверьте версию 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