Regarding cuda and cuDNN I think, correct me if I’m wrong, the problem is the same using pytorch. You have to register for cuDNN and have to match cuda and versions ☺