Jupyter Notebook、Colab安装导入pyautogui报KeyError: DISPLAY如何解决

报错原因 #

PyAutoGUI依赖图形化桌面显示环境才能运行，Google Colab的默认运行环境为无桌面的Linux服务器，未配置DISPLAY环境变量，因此导入PyAutoGUI时会触发该报错。

修复步骤 #

• 第一步：在Colab中安装PyAutoGUI、虚拟显示服务及对应Python调用库

!pip install pyautogui pyvirtualdisplay
!apt-get update
!apt-get install -y xvfb

• 第二步：在导入PyAutoGUI前启动虚拟显示服务

from pyvirtualdisplay import Display

# 启动指定分辨率的虚拟显示，visible=0表示不显示实体界面
display = Display(visible=0, size=(1920, 1080))
display.start()

# 此时再导入pyautogui即可正常运行
import pyautogui

注意事项 #

• 所有操作均在虚拟桌面中执行，无法直接查看界面效果，可调用pyautogui.screenshot('test.png')保存截图验证操作结果。
• 若需要停止虚拟显示，执行display.stop()即可。
• Colab运行环境与本地环境互相独立，你在本地Anaconda中安装的依赖无法直接在Colab中复用，所有需要的库都要在Colab中重新安装。

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