microsoft / evodiff

Generation of protein sequences and evolutionary alignments via discrete diffusion models
MIT License
516 stars 73 forks source link

UserWarning: Attempted to get default timeout for nccl backend, but NCCL support is not compiled #43

Open maggiedev2024 opened 2 months ago

maggiedev2024 commented 2 months ago

I am running this command on my macBook pro: python train.py /Users/username/Documents/projects/evodiff/config/config38M.json /Users/username/Documents/projects/evodiff/cat

Got this error, can anyone help to resolve it? Thanks a lot!

/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py:633: UserWarning: Attempted to get default timeout for nccl backend, but NCCL support is not compiled warnings.warn("Attempted to get default timeout for nccl backend, but NCCL support is not compiled") Traceback (most recent call last): File "/Users/username/Documents/projects/evodiff/train.py", line 454, in main() File "/Users/username/Documents/projects/evodiff/train.py", line 75, in main mp.spawn(train, nprocs=args.gpus, args=(args,)) File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/multiprocessing/spawn.py", line 282, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method="spawn") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/multiprocessing/spawn.py", line 238, in start_processes while not context.join(): ^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/multiprocessing/spawn.py", line 189, in join raise ProcessRaisedException(msg, error_index, failed_process.pid) torch.multiprocessing.spawn.ProcessRaisedException:

-- Process 0 terminated with the following error: Traceback (most recent call last): File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/multiprocessing/spawn.py", line 76, in _wrap fn(i, args) File "/Users/username/Documents/projects/evodiff/train.py", line 84, in train dist.init_process_group( File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/c10d_logger.py", line 79, in wrapper return func(args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/c10d_logger.py", line 93, in wrapper func_return = func(args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py", line 1368, in init_process_group defaultpg, = _new_process_group_helper( ^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py", line 1544, in _new_process_group_helper backend_config = BackendConfig(backend) ^^^^^^^^^^^^^^^^^^^^^^ File "/opt/anaconda3/envs/evodiff/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py", line 303, in init supported_devices = Backend.backend_capability[backend.lower()]


KeyError: 'nccl'