访问Keras / CuDNN自动调谐器中的TensorCores

时间:2018-04-02 12:57:41

标签: tensorflow keras cudnn

是否可以在Keras中启用CuDNN自动调谐器?在TensorFlow中,可以指定: os.environ['TF_CUDNN_USE_AUTOTUNE'] = "1" (目前默认启用)

我的目标是切换到支持Volta的TensorCores的CuDNN卷积算法。

0 个答案:

没有答案