Lorsque l’on commence à utiliser un GPU pour faire des calculs en PyTorch, on fait déjà un premier pas en avant vers des calculs optimisés. Pour autant, comment vérifier si l’on dispose d’un GPU sur lequel passer nos tenseurs ? Heureusement, PyTorch dispose d’une fonction spécialement conçue pour vérifier la disponibilité d’un GPU. Problème : … Lire la suite Vérifier la disponibilité d’un GPU avec PyTorch