部分解决cuda11导致的cudnn运行速度较cuda10慢的问题

318 阅读1分钟

问题如图,解决方案为调用
cudnnOpsInferVersionCheck(); cudnnCnnInferVersionCheck();
上述两个方法会创建全局cuda context,节约cuda context创建时间。