Vue d'ensemble
CuPy 11.0, publié le 15 juin 2022, apporte des améliorations de performance significatives et optimise l'utilisation de la mémoire GPU.
Fonctionnalités principales
Améliorations de performance
Les opérations élémentaires et les réductions sont accélérées grâce à une meilleure fusion de noyaux et un scheduling optimisé.
python
import cupy as cp
# Opérations accélérées
a = cp.random.randn(5000, 5000)
# Réduction optimisée
mean = cp.mean(a, axis=1)
std = cp.std(a, axis=1)
print(f'Forme moyenne: {mean.shape}') # (5000,)Gestion mémoire optimisée
Le pool de mémoire GPU est amélioré pour réduire la fragmentation et permettre une meilleure utilisation de la mémoire disponible.
python
import cupy as cp
# Gestion du pool mémoire
pool = cp.get_default_memory_pool()
print(f'Utilisée: {pool.used_bytes() / 1e6:.1f} MB')
print(f'Totale: {pool.total_bytes() / 1e6:.1f} MB')
# Libérer la mémoire inutilisée
pool.free_all_blocks()
print(f'Après libération: {pool.used_bytes() / 1e6:.1f} MB')