import MPI from mpi4py
parent
64f86b7aa6
commit
5feca7f4b6
|
@ -3,7 +3,7 @@ import time
|
||||||
import torch
|
import torch
|
||||||
import pickle
|
import pickle
|
||||||
import subprocess
|
import subprocess
|
||||||
|
from mpi4py import MPI
|
||||||
import torch.distributed as dist
|
import torch.distributed as dist
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue