GPU如何使用的元流解释
GPU如何使用的元流解释
提问于 2020-07-07 08:56:21
目标
了解GPU将如何在元流中使用。
背景
在关于如何使用GPU #250的文档/说明中，有几个关于如何使用GPU的讨论。
它看起来像@resources(GPU=2)负责GPU分配，但有一些讨论指出AWS EC2实例类型与GPU，如P或G实例，也是类型的AMI。
据我理解，Metaflow使用AWS批处理，它使用ECS/Docker。然后，码头实例需要在码头容器(NVIDIA容器工具包)内的GPU驱动程序能够访问GPU。
因此，不清楚元流如何管理GPU，以及使用GPU所需的先决条件、配置和编码是什么。
问题
- 我们是否需要使用在EC2实例(停靠主机)级别预配置GPU驱动程序的特定AMI？
- 我们是否需要在GPU中使用特定的EC2类型(P3、P4、G3、G4、INF1)？或者Metaflow是否使用服务，例如AWS弹性推理来动态分配GPU，即使批/ECS EC2实例没有GPU？
- Metaflow是在码头容器内安装GPU驱动程序还是在内部使用NVIDIA容器工具包？
- @资源是我们在Python代码中使用GPU所需要的全部吗？
回答 1
Stack Overflow用户
发布于 2020-07-17 06:44:44
在元流中使用GPU。
- 需要一个DL AMI或自定义AMI有NVIDIA驱动程序安装。使用AWS DL AMI。
- 确保AWS批处理在计算环境配置中具有EC2 p2/p3实例。
- 元流不负责GPU驱动程序和库，但是
metaflow configure aws允许使用哪个坞映像。指定已安装NVIDIA驱动程序和CUDA的坞映像(ECR存储库)。使用AWS深度学习容器。 - 使用@Batch和specif属性。
吉特布事件的细节。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/62771540
复制相关文章
相似问题
腾讯云开发者
