Skip to content

Commit 8222dc9

Browse files
Revert "Update ddp_spawn.py"
This reverts commit f172101.
1 parent 6c095b2 commit 8222dc9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_lightning/plugins/training_type/ddp_spawn.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import torch.multiprocessing as mp
2222
from torch.nn.parallel.distributed import DistributedDataParallel
2323
from torch.optim import Optimizer
24+
import numpy
2425

2526
from pytorch_lightning.distributed.dist import LightningDistributed
2627
from pytorch_lightning.overrides import LightningDistributedModule
@@ -78,6 +79,7 @@ def distributed_sampler_kwargs(self):
7879

7980
def setup(self, model):
8081
os.environ["MASTER_PORT"] = str(self.cluster_environment.master_port())
82+
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
8183

8284
# pass in a state q
8385
smp = mp.get_context("spawn")

0 commit comments

Comments
 (0)