问题遇到的现象和发生背景
pytorch中设置torch.backends.cudnn.benchmark = True时,cudnn会自动选择最快的卷积算法,我想弄清cudnn在pytoch的卷积优化中究竟使用了什么算法(FFT?GEMM?WINOGRAD),于是想在网络运行时同步导出cudnn的日志,但是使用pytorch卷积层Conv3d时,导出的cudnn日志文件是空的
用代码块功能插入代码,请勿粘贴截图
按照cudnn官网的说明,已经在环境变量中配置 CUDNN_LOGINFO_DBG = 1,CUDNN_LOGWARN_DBG=1, CUDNN_LOGERR_DBG=1 以及 CUDNN_LOGDEST_DBG = C:\Users\Administrator\Desktop\cudnn_log\log.txt。
运行结果及报错内容
在指定位置生成了日志文件,但是是空的