引言
随着这几年深度学习的大火,入了坑才发现 机器学习虽然智能,但是调参是个很大的坑。得益于这几年工具的发展,autoML有了进展。炼丹师们有救了。
本文就以nni库为例,来说明具体如何设置gpu。在nni的introduction中直接忽略了这部分,造成了配置文件中虽然设置了gpuNum:1,nvidia-smi操作后发现后台几乎没有调用gpu。去查日志文件才发现:gpu_metrices file not exit.
prerequisite
测试环境: Ubuntu 16.04
所需安装库: cuda, tensorflow, nni
解决方案
在配置文件中,加入如下一行:
localConfig:
useActiveGpu: true
完整配置文件如下:
先去拷贝库的源代码,找到example部分
git clone -b v1.5 https://github.com/Microsoft/nni.git
cd 进入 ./nni/examples/trials/mnist-annotation目录
mnist-config_gpu.yml文件如下所示:
authorName: default
experimentName: example_mn