# 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()