By using AWS re:Post, you agree to the Terms of Use

Use Nvidia GPU in AWS Batch


For a job, I run puppeteer in the headful mode in AWS Batch. I found a docker file that uses the image node:16.5-slim, then downloads Chrome, and then uses Xvfb.

If I run it on an EC2 instance with GPU, Chrome still uses the integrated video card (llvmpipe) not taking advantage of the powerful Nvidia GPU.

I found online a couple of attempts, one I heard they had to switch to Windows (not supported by AWS Batch). They also used an AMI only available on EC2:

And the other solution also doesn't seem to be suitable because of its complexity to set up:

I wonder if there is a docker image I could use as a starting point where the Nvidia GPU is already set up and Chrome could take advantage of it with only a few extra configurations. Thanks!

asked 6 months ago50 views
No Answers

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions