标签: caffe
我正在尝试调试导致崩溃的问题,多线程使我的故障排除变得复杂。如何在Visual Studio中为Caffe的Windows分支关闭多线程?
答案 0 :(得分:1)
从v1.0.1开始,torch.set_num_threads(1)涵盖了所有内容,包括BLAS库。 我在PyTorch 1.6和Python 3.6上进行了测试,它总共收到3个线程(其中1个用于Python)。
torch.set_num_threads(1)
来源:https://github.com/pytorch/pytorch/issues/16899