如何设置MXNET_CUDNN_AUTOTUNE_DEFAULT?

时间:2017-02-21 00:36:53

标签: deep-learning mxnet

当我测试图像时,在mxnet enter image description here

中使用更快的R-CNN

如何设置MXNET_CUDNN_AUTOTUNE_DEFAULT?

1 个答案:

答案 0 :(得分:2)

如果在linux或unix中使用mxnet,则应设置环境 "导出MXNET_CUDNN_AUTOTUNE_DEFAULT = 1"。 如果在Windows中使用mxnet,则应设置环境 "设置MXNET_CUDNN_AUTOTUNE_DEFAULT = 1"。

请参阅http://mxnet.io/how_to/perf.html