Exécutez du code python avec GPU

GPU exécutant du code python

étape

1. Vérifiez la carte graphique de l'ordinateur (entrée dans la console)

nvidia-smi

Le contenu suivant apparaît.

Il n'y a que la carte graphique n ° 0

2. Spécifiez le GPU dans le code

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

C'est ça! ! ! !