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。
在指定位置生成了日志文件,但是是空的
你是在系统变量还是用户变量?系统变量需要重启一下explore.exe(windows 资源管理器),甚至有写需要重启电脑才能生效,直接cmd中echo不会生效。
重启不行的话,试试通过在代码中添加环境变量配置呢?你试下下面的是否可行
import os
os.environ["CUDNN_LOGINFO_DBG"] = '1'
os.environ["CUDNN_LOGWARN_DBG"] = '1'
os.environ["CUDNN_LOGERR_DBG"] = '1'
os.environ["CUDNN_LOGDEST_DBG "] =‘C:/Users/Administrator/Desktop/cudnn_log/log.txt’