GPU在docker和kubernetes中的使用

伴随着人工智能、机器学习、深度学习等技术的火热,GPU近年来也得到了快速的发展。GPU可以大大加快深度学习任务的运行速度。而像Tensflow这样的框架的出现和应用更是离不开对GPU资源的依赖。同时,GPU资源又是十分昂贵的,需要尽可能提高GPU资源的利用率。为了解决上述问题,我们利用Kubernetes将GPU资源聚合成资源池来实现统一管理,并借用Docker交付深度学习的运行时环境。当前kubernetes中还只支持NvidiaGPU。所以,本文以NvidiaGPU来进行说明。docker本身并不原生支持GPU,但是使用docker的现有功能可以对GPU的使用进行支持。 如上所述,通过--