recommenders-team/recommenders

[ASK] How to run with docker in GPU environment

Open

#1,526 opened on Sep 16, 2021

View on GitHub
 (12 comments) (0 reactions) (0 assignees)Python (2,972 forks)batch import
help wanted

Repository metrics

Stars
 (17,706 stars)
PR merge metrics
 (Avg merge 6d 16h) (10 merged PRs in 30d)

Description

I follow the start-up of docker in GPU environment like this: image I succeed in running it. However when I check the GPU-util, it appears that it doesn't run on GPU. image How can I use my GPU?

Contributor guide

Research direction
Verify that the NVIDIA Container Toolkit is installed and configured. Ensure Docker container is run with gpus all flag. Inside container, run nvidia smi to check GPU visibility. Check that the application is using GPU by monitoring with nvidia smi during execution.
Tech stack
pythondocker
Domain
backenddevops
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.
1-3 hours
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
DockerNVIDIA Container Toolkit
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
65