PyTorch torch.set_num_interop_threads Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.set_num_interop_threadsDiese Funktion in PyTorch wird verwendet, um die Anzahl der inter-op-parallelen Threads festzulegen. Diese Threads werden verwendet, um unabhängige, unterschiedliche Operationen parallel auszuführen.

Funktionsdefinition

torch.set_num_interop_threads(num_threads)

Parameterbeschreibung

  • num_threads: Die Anzahl der festzulegenden Inter-op-Threads

Anwendungsbeispiel

Beispiel

import torch

# Anzahl der Inter-op-Threads festlegen
torch.set_num_interop_threads(4)

print("Die Anzahl der Inter-op-Threads wurde festgelegt auf:", torch.get_num_interop_threads())

# Standardwert wiederherstellen
torch.set_num_interop_threads(1)

print("Standardanzahl der Inter-op-Threads wiederhergestellt")

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen