Crée un accélérateur

🌐

Create an accelerator

Crée un accélérateur

Utilisation

accelerator(
  device_placement = TRUE,
  cpu = FALSE,
  cuda_index = torch::cuda_current_device()
)

Arguments

device_placement

(booléen) si l'objet accelerator doit gérer le placement sur les cartes accélératrices. Par défaut : TRUE

cpu

(booléen) le processus d'apprentissage doit-il s'exécuter sur le processeur central ?

cuda_index

(entier) index de la carte CUDA à utiliser si plusieurs GPUs sont disponibles. Par défaut : le résultat de torch::cuda_current_device()