La Data Science

Tout sur la data

Vérifier la disponibilité d’un GPU avec PyTorch

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 :

Comment savoir si j’ai un GPU disponible à utiliser avec PyTorch?

Solution :

Utiliser la fonction is_available de torch.cuda

Exemple d’utilisation :

	import torch

	is_free = torch.cuda.is_available()
	if is_free:
	print("GPU disponible.")
	else:
	print("Pas de GPU disponible.")

view raw GPU-available.py hosted with ❤ by GitHub

Jean MP