Sagemaker Notebook ml.g4dn.xlarge 实例不列出 GPU 设备

Question

我使用 amazonei_tensorflow2_p36 内核创建了一个实例类型为“ml.g4dn.xlarge”的 sagemaker 笔记本。对于 tensorflow 版本 2.3.2，tf.config.list_physical_devices('GPU') 返回 PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')

但是，当我将 tensorflow 版本更改为 2.4.1（使用 !pip install --upgrade pip && pip install tensorflow==2.4.1 && pip install tensorflow-gpu==2.4.1 安装）时，tf.config .list_physical_devices('GPU') 返回 []

如何在 tensorflow 2.4.1 版本中使用 GPU？

Answer 1

我在文档中找不到任何参考资料，但我怀疑 Sagemaker 并不完全支持您更新 Tensorflow 的方式。 Sagemaker 通常使用预先构建的容器，这些容器已经包含 tensorflow，但也包含 CUDA。如果您只是更新 Tensorflow，则无法保证仍与已安装的 CUDA 兼容。