如何使用/激活amazon ami深度学习使用GPU而不是CPU

Question

我最近决定尝试亚马逊aws ce2 - 亚马逊深度学习ami - 3a533040。 但是，在连接到机器并更新库之后，我仍然无法弄清楚如何在GPU上运行。

我正在使用GPU计算实例，并且我正在激活源tensorflow_p36。 然而，当我运行代码（使用Conda配置）时，它总是在CPU上运行，而不是在GPU上运行。

Answer 1

您使用哪种实例类型启动实例？ 您的代码还会在gpu设备上下文中执行吗？
您也可以访问AWS Developer论坛以获取社区https://forums.aws.amazon.com/forum.jspa?forumID=263

的帮助。