replicate/cog

Warn if nvidia-docker is not installed

Ouverte

#625 ouverte le 27 mai 2022

Voir sur GitHubGuide contributeur
 (2 commentaires) (3 réactions) (0 personne assignée)Python (356 forks)batch import
documentationgogood first issue

Métriques du dépôt

Stars
 (5 724 étoiles)
Métriques de merge PR
 (Merge moyen 7j 8h) (19 PRs mergées en 30 j)

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.

Guide contributeur

Direction de recherche
Le problème demande une vérification au début des exécutions du modèle GPU. Étudiez la gestion actuelle des erreurs dans le code Python de cog, probablement dans des commandes comme 'cog predict' ou 'cog run'. Recherchez où Docker est invoqué et ajoutez une pré vérification en utilisant subprocess pour tester nvidia docker. Fournissez un message d'erreur utile avec des instructions d'installation.
Stack technique
pythondocker
Domaine
backenddevops
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
Dockernvidia docker
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
70