标签: tensorflow keras cudnn
是否可以在Keras中启用CuDNN自动调谐器?在TensorFlow中,可以指定: os.environ['TF_CUDNN_USE_AUTOTUNE'] = "1" (目前默认启用)
os.environ['TF_CUDNN_USE_AUTOTUNE'] = "1"
我的目标是切换到支持Volta的TensorCores的CuDNN卷积算法。
TensorCores