Я сам решил это следующим образом:
Сначала запустите это в терминале
sudo apt-get install xvfb
sudo pip3 install pyvirtualdisplay
Затем добавьте следующие строки в ваш код
from pyvirtualdisplay import Display
display = Display(visible=0,size=(1024,768))
display.start()
И моя конфигурация браузера выглядит так:
cap = DesiredCapabilities().FIREFOX
cap["marionette"] = False
display = Display(visible=0,size=(1024,768))
display.start()
options = Options()
options.set_headless(headless=True)
binary = FirefoxBinary("/home/ubuntu/firefox/firefox")
options.add_argument("-headless")
browser = Firefox(firefox_options=options, executable_path='/home/ubuntu/Documents/sourcecode/geckodriver',firefox_binary=binary,capabilities = cap )