Fonction torch.set_num_interop_threads de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.set_num_interop_threadsCette fonction de PyTorch est utilisée pour définir le nombre de threads parallèles inter-op. Ces threads sont utilisés pour exécuter en parallèle différentes opérations indépendantes.

Définition de fonction

torch.set_num_interop_threads(num_threads)

Description du paramètre

  • num_threads: Le nombre de threads inter-op à définir

Exemple d'utilisation

Exemple

import torch

# Définir le nombre de threads inter-op
torch.set_num_interop_threads(4)

print("Le nombre de threads inter-op a été défini à :", torch.get_num_interop_threads())

# Restaurer la valeur par défaut
torch.set_num_interop_threads(1)

print("Nombre de threads inter-op par défaut restauré")

Le résultat de sortie est:

跨域线程数已设置为: 4
已恢复默认跨域线程数

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions