facebookresearch/MLGym

GPU requirements on Docker

Open

#10 opened on Mar 5, 2025

View on GitHubContributor guide
 (3 comments) (0 reactions) (0 assignees)Python (59 forks)auto 404
enhancementgood first issuehelp wantedtasks

Repository metrics

Stars
 (612 stars)
PR merge metrics
 (PR metrics pending)

Description

Hello,

I noticed in the setup that the docker containers are expected to have GPU access. This is verified with nvidia-smi.

I have 2 questions:

  1. Is GPU access required to be able to properly evaluate on this benchmark?
  2. How should a mac user with a non-nvidia gpu go about the setup?

Thanks in advance!

Contributor guide

Research direction
Investigate if GPU is mandatory for MLGym benchmarks. Explore CPU only fallback options and Mac friendly alternatives (e.g., using Metal API via Docker with GPU passthrough or cloud GPU). Check official docs for non NVIDIA GPU support.
Tech stack
python
Domain
backenddevopsinfrastructuremachine learningai
Issue type
Research
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
Under 1 hour
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
Docker
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
65