如何在Sagemaker的笔记本实例中连接到数据库？

【以下的回答经过翻译处理】 您好，

一般原则是笔记本实例和数据库实例之间必须有网络连接，数据库实例的安全组应允许来自笔记本实例的入站流量。

其中一个设置的示例为：

1. RDS数据库实例位于VPC vpc-a和子网subnet-b。
2. SageMaker笔记本在VPC vpc-a中启动，在子网subnet-b中，使用安全组sg-c并启用了“禁用直接连接互联网”的选项。
3. 在RDS数据库实例的安全组规则中，您可以添加一个入站规则，允许来自SageMaker笔记本安全组“sg-c”的入站流量。

-- 类型 - 协议 - 端口范围 - 来源

-- MYSQL/Aurora - TCP - 3306 - sg-c

示例代码：

！pip install mysql-connector

import mysql.connector
mydb = mysql.connector.connect(
host=”$RDS_ENDPOINT”,
user=”$RDS_USERNAME”,
passwd=”$RDS_PASSWORD”
)
cursor = mydb.cursor()
cursor.execute(“SHOW DATABASES”)

感谢您使用Amazon SageMaker，如果有其他需要帮助的地方，请随时联系我们！