运行MXNET时,默认会进行autotune:
[16:40:49] src/operator/nn/./cudnn/./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable)
可以输入一下命令来避免:在shell文件中执行Python程序前加上:
export MXNET_CUDNN_AUTOTUNE_DEFAULT=0

本文介绍如何在运行MXNet时禁用CUDNN的自动调优功能,以避免运行性能测试导致的时间消耗。通过设置环境变量MXNET_CUDNN_AUTOTUNE_DEFAULT为0,可以在执行Python程序前避免这一过程。
2435





