replicate/cog

Warn if nvidia-docker is not installed

Open

#625 opened on May 27, 2022

 (2 comments) (3 reactions) (0 assignees)Python (356 forks)batch import
documentationgogood first issue

Repository metrics

Stars
 (5,724 stars)
PR merge metrics
 (Avg merge 7d 8h) (19 merged PRs in 30d)

Description

When nvidia-docker isn't installed and you're trying to run a GPU model, Cog blows up with a big unparseable stack trace. We should check for nvidia-docker and show instructions for how to install nvidia-docker if it's not available.

Contributor guide