Openshift,tensorflow和GPU

时间:2018-03-03 06:42:44

标签: tensorflow kubernetes openshift minishift

我想使用OpenShift来构建和部署我的Tensorflow应用程序。我也想使用GPU运行/服务于tensorflow应用程序。

我很欣赏Kubernetes和Openshift中的GPU支持相当新,因此文档很少。

我尝试在minishift上运行https://github.com/radanalyticsio/tensorflow-serving-gpu-s2i但是无法使其正常工作。它失败了

0/1 nodes are available: 1 insufficient alpha.kubernetes.io/nvidia-gpu
  • minishift支持GPU使用吗?
  • 支持哪些GPU卡 Kubernetes(从而OpenShift和希望MiniShift)?
  • 如何分配OpenShift项目以使用特定的图形卡?在Minishift中创建TF服务项目时,使用上面的链接,可以选择命名GPU。然而,Tensorflow - 据我所知 - 按编号使用GPU卡。

1 个答案:

答案 0 :(得分:0)

正如@mon在评论中所说,我还建议你浏览这两篇博文:

在节点上,需要安装驱动程序和CUDA。如果我没有弄错的话,nvidia-docker被用在封面下。它应该自动将共享库安装到每个新容器中。但是,他们发布的v2.0与1.x基本不同。

我相信,一旦你准备好了OpenShift节点,那么只有s2i这样的其他功能才能帮助你创建图像。

你也可以找到这些Subin的要点: