Nvidia hisse CUDA Python 1.0 au niveau de C++

Une seule commande doit suffire pour installer le nouvel environnement Python : pip install cuda-python cuda-cccl numba-cuda-mlir[cu13]. Nvidia publie CUDA Python 1.0 avec CUDA 13.3 et positionne cette interface au même niveau que C++.

Le cœur du système s’appelle cuda.core. Il fournit aux programmes Python des objets correspondant aux appareils CUDA, aux streams — autrement dit, des flux d’exécution — et à la mémoire. Les développeurs peuvent ainsi écrire leurs propres kernels, de petits programmes dédiés aux calculs parallèles, ou optimiser des applications existantes.

Le paquet regroupe plusieurs composants : cuda.compute fournit des algorithmes parallèles comme le tri, la recherche et les histogrammes, tandis que cuda.bindings relie directement Python à toutes les interfaces C. cuda-pathfinder détecte les composants CUDA installés et nvmath-python apporte des bibliothèques mathématiques à Python. Selon Nvidia, les paquets jusque-là utilisés comme PyTorch, CuPy ou RAPIDS pourraient ainsi devenir inutiles.

D’après Nvidia, la différence technique réside dans le modèle d’objet commun. Plusieurs composants et applications accèdent aux mêmes objets CUDA sous-jacents, qui peuvent ainsi être gérés au-delà des frontières entre bibliothèques. Cela doit également s’appliquer aux Green Contexts, c’est-à-dire aux kernels bénéficiant d’avantages particuliers en matière de latence.

Et ensuite ? Selon Nvidia, les développeurs Python disposent d’un accès plus direct à CUDA, sans devoir impérativement utiliser des bibliothèques intermédiaires supplémentaires pour accéder aux fonctions C++. Nvidia promet pour cela une API stable, reposant sur une gestion sémantique des versions : les changements incompatibles ne devraient apparaître que dans les versions majeures, tandis que les fonctions seront supprimées selon des règles prévisibles. L’utilisation de l’environnement publié devra montrer si cette uniformisation promise remplace effectivement les bibliothèques existantes au quotidien.