Función PyTorch torch.set_num_interop_threads
Manual de referencia de Pytorch torch
torch.set_num_interop_threadsEs una función en PyTorch que se utiliza para establecer el número de hilos de paralelismo entre operaciones (inter-op). Estos hilos se utilizan para ejecutar diferentes operaciones independientes en paralelo.
Definición de la función
torch.set_num_interop_threads(num_threads)
Descripción de parámetros
num_threads: el número de hilos inter-op que se desea establecer
Ejemplo de uso
Ejemplo
import torch
# Establecer el número de hilos inter-op
torch.set_num_interop_threads(4)
print("El número de hilos inter-op se ha establecido en:", torch.get_num_interop_threads())
# Restaurar el valor predeterminado
torch.set_num_interop_threads(1)
print("Se ha restaurado el número predeterminado de hilos inter-op")
# Establecer el número de hilos inter-op
torch.set_num_interop_threads(4)
print("El número de hilos inter-op se ha establecido en:", torch.get_num_interop_threads())
# Restaurar el valor predeterminado
torch.set_num_interop_threads(1)
print("Se ha restaurado el número predeterminado de hilos inter-op")
El resultado de salida es:
跨域线程数已设置为: 4 已恢复默认跨域线程数
Otras extensiones