Vaincre l'enfer de l'installation multiplateforme de PyTorch

2025-09-11

Créer un projet Python multiplateforme qui repose sur PyTorch est notoirement difficile. Lors du développement de FileChat, un assistant de codage IA, l'auteur a rencontré ce défi. La gestion des dépendances standard perd des index personnalisés lors de la création de paquets wheel de distribution, nécessitant une configuration manuelle de l'utilisateur. En tirant parti de PEP 508, l'auteur a spécifié des URL de wheel pour chaque dépendance ainsi que des contraintes de version Python, permettant une installation en une seule commande. Windows et macOS utilisent le PyTorch par défaut, tandis que Linux propose des wheels séparées pour le matériel CPU, XPU et CUDA. Les utilisateurs sélectionnent le groupe de dépendances optionnelles approprié lors de l'installation (par exemple, `pip install filechat[xpu]`). Maintenir les URL des wheels est plus simple que de gérer des index personnalisés, bien que cela nécessite plus de travail préalable.

Développement