运行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