Amoozegar
New Contributor II

Also I should mention I'm using Databricks Runtime 10.4 for ML, and my assumption is that I might not have direct control over upgrading CUDA or cuDNN within the runtime environment.