# pip install pyvirtualdisplay # pip install selenium from pyvirtualdisplay import Display from selenium import webdriver # 启动虚拟屏幕 display = Display(visible=0, size=(800, 600)) display.start() # 创建浏览器实例并访问页面 browser = webdriver.Chrome() browser.get('https://www.baidu.com') # 截图并保存 browser.save_screenshot('screenshot.png') # 关闭浏览器 browser.quit() # 关闭虚拟屏幕 display.stop()