Passer au contenu principal
Pour les jobs W&B Launch qui utilisent un accélérateur, fournissez une image de base qui inclut les composants d’accélération nécessaires. Cela garantit que votre job dispose des pilotes GPU requis, des bibliothèques d’exécution et des autres dépendances matérielles nécessaires pour s’exécuter. L’image de l’accélérateur doit respecter les exigences suivantes :
  • Compatibilité avec Debian (le Dockerfile de Launch utilise apt-get pour installer Python).
  • Prise en charge des jeux d’instructions matériels CPU et GPU (confirmez la compatibilité de la version CUDA avec le GPU cible).
  • Compatibilité entre la version de l’accélérateur fournie et les paquets de l’algorithme de machine learning.
  • Installation des paquets qui nécessitent des étapes supplémentaires pour assurer la compatibilité matérielle.