python - 在 AWS Sagemaker 中启用 gpu 支持

Question

我正在尝试在 Sagemaker 笔记本上启用 GPU。虽然我使用的是 ml.p2.xlarge 实例，但当我在我的笔记本中尝试这个时:

import tensorflow as tf
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)

输出为False。没有 GPU，我的训练过程会很慢。如何在我的 Sagemaker 笔记本中激活和使用 GPU？

Answer 1

我假设您在 Notebook 中使用的是 _amazonei_ 内核？尝试使用不同的内核，例如 conda_tensorflow*

名称中包含 amazonei 的内核与 GPU 不兼容，它们用于 Amazon Elastic Inference Accelerator.

• 注意 1 P2 使用老一代 K80 GPU；您可以在 SageMaker Studio 中访问更新(和更便宜)的东西:G4笔记本(有T4卡)

• 注释 2 根据 TF documentation ，您使用的功能已弃用。一旦你在一个 GPU 兼容的内核上，TF 推荐使用 tf.config.list_physical_devices('GPU')