在GoogleColab运行含PyAutoGui的Python脚本时遇DISPLAY错误

错误原因 #

PyAutoGui依赖的mouseinfo库需要读取系统的DISPLAY环境变量来连接X11显示服务器，而Google Colab的运行环境是无图形界面的云端服务器，默认没有配置X11显示环境，也未设置DISPLAY变量，因此导入时会触发KeyError: 'DISPLAY'。

注：日志开头的cuDNN/cuFFT/cuBLAS注册警告属于TensorFlow的环境冲突提示，和本次PyAutoGui的崩溃问题无关，可以忽略。

解决方法 #

通过在Colab中创建虚拟X11显示环境来解决，具体操作步骤：

• 安装虚拟显示依赖包
  在Colab的代码单元格中执行以下命令：

  !apt-get install -y xvfb x11-utils
  !pip install pyautogui mouseinfo
  

• 配置并启动虚拟显示
  在导入PyAutoGui的代码前添加以下初始化逻辑：

  import os
  import subprocess
  import time
  
  # 启动虚拟显示服务器
  subprocess.Popen(['Xvfb', ':1', '-screen', '0', '1024x768x16'])
  # 设置DISPLAY环境变量指向虚拟显示
  os.environ['DISPLAY'] = ':1'
  # 等待虚拟显示服务初始化完成
  time.sleep(2)
  
  # 此时可正常导入并使用PyAutoGui
  import pyautogui
  

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