크롬 Headless 모드에서 GPU 가속하기
크롬 브라우저의 headless 모드는 웹사이트 테스트나 웹페이지 크롤링에 주로 사용된다. 이때, 웹페이지를 png나 pdf로 저장하기도 하기도 하고, WebGL과 같은 3D 컨텐츠를 테스트할 필요도 있다. 늘 그렇지만 기본적으로 3D 가속은 되지 않는다. 때로는 가속되다가도 크롬 브라우저가 버전업하면서 안되기도 한다. 당연히 버그도 이미 올라와있고, 이 글을 쓰는 본인도 약간의 기여를 한적은 있다. :-)
최근 구글 크롬 개발자 블로그에서 헤드리스 모드에서 GPU 가속을 활성화하는 방법을 자세히 공개했다.
https://developer.chrome.com/blog/supercharge-web-ai-testing
웹 AI 모델 테스트 강화: WebGPU, WebGL, 헤드리스 Chrome | Blog | Chrome for Developers
브라우저 테스트를 지속적으로 자동화하고 애플리케이션 성능을 개선하는 방법에 대한 Google Cloud 솔루션을 살펴보세요.
developer.chrome.com
사실 리눅스용 크롬 브라우저에서는 GPU 가속이 꺼져있다. 리눅스는 distro도 다양하고 게다가 GPU벤더가 지원하는 공식 GPU 드라이버에서 해커들이 개발한 오픈소스 버전도 있기 때문에 구글에서 GPU 가속을 기본적으로 꺼놨다. 현재도 마찬가지로 일부 distro는 자체 GPU 가속이 되는 크롬 브라우저 패키지를 내놓고도 있거나, GPU 가속 방법을 알려주고 있다.
아래 코드를 실행하려면 먼저 GPU driver를 설치되고 잘 동작해야 한다. 아래는 우분투 22.04에서 동작하는 NVidia GPU 예이다.
$ nvidia-smi
Wed Jan 24 10:32:13 2024
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 545.29.06 Driver Version: 545.29.06 CUDA Version: 12.3 |
|-----------------------------------------+----------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+======================+======================|
| 0 NVIDIA GeForce RTX 4090 Off | 00000000:01:00.0 Off | Off |
| 0% 45C P8 13W / 450W | 1078MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
+---------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=======================================================================================|
| 0 N/A N/A 4265 G /usr/bin/gnome-shell 6MiB |
| 0 N/A N/A 42318 C /usr/local/bin/ollama 1056MiB |
+---------------------------------------------------------------------------------------+아래 코드는 크롬 브라우저의 GPU 환경 페이지를 pdf로 출력해주는 코드이다.
/* Complete example.js */
import puppeteer from 'puppeteer';
// Configure launch parameters
const browser = await puppeteer.launch({
headless: 'new',
args: [
'--no-sandbox',
'--headless=new',
'--use-angle=vulkan',
'--enable-features=Vulkan',
'--disable-vulkan-surface',
'--enable-unsafe-webgpu',
]
});
const page = await browser.newPage();
await page.goto('chrome://gpu');
// Verify: log the WebGPU status or save the GPU report as PDF
const txt = await page.waitForSelector('text/WebGPU');
const status = await txt.evaluate(g => g.parentElement.textContent);
console.log(status);
await page.pdf({path: './gpu.pdf'});
await browser.close();크롬 브라우저를 직접 실행하여 'chrome://gpu' 페이지를 확인하고, 위 코드로 생성된 PDF 결과가 동일하다면 헤드리스 모드에서 GPU 가속이 지원되는 것으로 볼 수 있다.
'Chromium' 카테고리의 다른 글
| 크로미엄(Chromium) 프로젝트 기여하기 (2) | 2021.05.01 |
|---|---|
| 크롬대학(Chrome University) 2018: Chrome OS Graphics 101 (0) | 2020.05.09 |