部署在ECR的AWS Lambda中ChromeDriver+Selenium启动失败问题

1. 适配Lambda的Docker镜像构建 #

Lambda的容器环境基于Amazon Linux，需用官方兼容的基础镜像构建，同时正确安装Chrome、ChromeDriver并处理依赖权限。修改后的Dockerfile示例：

# 基于Lambda官方Python镜像
FROM public.ecr.aws/lambda/python:3.10

# 安装Chrome运行依赖及浏览器本体
RUN yum install -y unzip wget xorg-x11-server-Xvfb libX11 libXext libXrender libXtst libXScrnSaver
RUN wget https://dl.google.com/linux/direct/google-chrome-stable_current_x86_64.rpm
RUN rpm -i google-chrome-stable_current_x86_64.rpm

# 下载与Chrome版本匹配的ChromeDriver
RUN CHROME_VERSION=$(google-chrome --version | grep -oP '\d+\.\d+\.\d+\.\d+') && \
    wget https://chromedriver.storage.googleapis.com/$CHROME_VERSION/chromedriver_linux64.zip && \
    unzip chromedriver_linux64.zip && \
    mv chromedriver /usr/bin/chromedriver && \
    chmod +x /usr/bin/chromedriver

# 安装Python依赖
COPY requirements.txt .
RUN pip install -r requirements.txt

# 复制业务代码到Lambda工作目录
COPY your_script.py ${LAMBDA_TASK_ROOT}

# 设置Lambda执行入口
CMD ["your_script.lambda_handler"]

2. 调整Chrome启动参数适配Lambda环境 #

Lambda无图形界面，必须启用无头模式，且需添加一系列兼容参数规避环境限制。修改Python代码中的Chrome配置：

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

def lambda_handler(event, context):
    chrome_options = Options()
    # 新版无头模式，行为更接近正常Chrome
    chrome_options.add_argument("--headless=new")
    # Lambda环境不支持沙箱，必须禁用
    chrome_options.add_argument("--no-sandbox")
    # 避免/dev/shm内存不足问题
    chrome_options.add_argument("--disable-dev-shm-usage")
    # 禁用GPU加速（无头模式无需）
    chrome_options.add_argument("--disable-gpu")
    # 设置固定窗口大小，避免页面布局异常
    chrome_options.add_argument("--window-size=1920x1080")
    # 指定配置文件路径到Lambda唯一可写目录/tmp
    chrome_options.add_argument("--user-data-dir=/tmp/chrome-profile")

    # 加载已登录的Chrome配置文件（需确保配置文件已复制到/tmp且权限正确）
    # 若将配置文件打包进镜像，需添加：RUN chmod -R 777 /tmp/chrome-profile

    # 初始化WebDriver
    driver = webdriver.Chrome(options=chrome_options)

    # 执行邀请链接处理逻辑
    driver.get("提取到的Snapchat邀请链接")
    # ... 后续接受邀请的操作 ...

    driver.quit()
    return {"status": "invite accepted"}

3. 处理Chrome配置文件的权限与路径 #

• 已登录的Chrome配置文件必须放在/tmp目录（Lambda仅该目录可写），不能放在镜像的其他只读路径。
• 若将配置文件打包进镜像，需添加RUN chmod -R 777 /tmp/chrome-profile，避免Lambda运行时权限不足。
• 若配置文件体积较大，可上传至S3，Lambda启动时先下载到/tmp再加载。

4. 调整Lambda资源配置 #

Chrome和WebDriver需要足够内存支持，建议：

• 将Lambda内存设置为至少1024MB（2048MB更佳）
• 超时设置为至少30秒，避免启动或操作过程中超时

5. 本地调试方法 #

• 在Dockerfile中添加版本验证命令：RUN google-chrome --version和RUN chromedriver --version，确保版本匹配。
• 本地运行镜像模拟Lambda环境：docker run -p 9000:8080 your-image，再用curl -XPOST "http://localhost:9000/2015-03-31/functions/function/invocations" -d '{}'测试，查看是否能正常启动Chrome。
• 在Python代码中添加日志打印，比如print(driver.capabilities)，确认WebDriver是否成功初始化。

内容的提问来源于stack exchange，提问作者beni14