Sagemaker笔记本ml.g4dn.xlarge实例未列出GPU设备

Question

我用amazonei_tensorflow2_p36内核创建了一个实例类型为‘ml.g4dn.xlarge’的sagemaker笔记本。在tensorflow版本2.3.2中，tf.config.list_physical_devices('GPU')返回PhysicalDevice(名称=‘/物理设备:GPU:0’，设备类型=‘GPU’)

但是，当我将tensorflow版本更改为2.4.1 (随!pip安装--升级pip && pip安装tensorflow==2.4.1 && pip install tensorflow-GPU==2.4.1安装)时，tf.config.list_physical_devices('GPU')返回[]

如何在tensorflow 2.4.1版中使用GPU？

Answer 1

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