Prérequis matériels

Les paquets ML Document Understanding 2021.10 sont différents des versions précédentes dans la manière dont ils utilisent les GPU et les CPU.

Les versions précédentes formaient la même architecture de modèle, que le pipeline d'entraînement soit exécuté sur un GPU ou sur un CPU. Cependant, les paquets ML Document Understanding 2021.10 incluent une optimisation qui réduit la taille du modèle lors de la formation sur le processeur, afin d'accélérer le processus de formation.

En conséquence, l'entraînement sur GPU n'est que 5 fois plus rapide que sur CPU (auparavant, il était 10 à 20 fois plus rapide). Cela permet également de former des modèles sur CPU avec jusqu'à 5000 pages (auparavant, 500 maximum).

Veuillez noter que l'entraînement de modèles de Document Understanding sur GPU nécessite un GPU avec au moins 11 Go de RAM vidéo pour fonctionner correctement.

Utilisez la table ci-dessous pour vérifier la compatibilité entre les paquets ML, la version CUDA et la version du pilote GPU.

Version du paquet ML	Version CUDA	version cudDNN	Pilote NVIDIA (version compatible la plus ancienne)	Génération de matériel
2021.10	CUDA 11.1	cuDNN 8.0.5	R450.36.06	Ampère, Turing, Volta, Pascal, Maxwell, Kepler

CUDA est rétrocompatible, ce qui signifie que les applications CUDA existantes peuvent continuer à être utilisées avec les versions plus récentes de CUDA.

Vous trouverez plus d'informations sur la personnalisation ici.