代码似乎只能在3个gpu的单机运行才行。否则出错。能否改成在单gpu单机上运行？

[springfall2018]

您好。我发现您们的代码似乎只能在3个gpu的单机运行才行。否则出错
例如我在google colab上运行
!python fed_seed_run.py /content/drive/MyDrive/workspace/ fedavg mrpc adapter 1100 0，1，3

训练相当长时间，报错收不到客户消息超时退出。想改成!python fed_seed_run.py /content/drive/MyDrive/workspace/ fedavg mrpc adapter 1100 0 运行也报错
因为colab只能有一个gpu,不知您有没有办法让他在单gpu机器上运行？（或哪里可找到多gpu的云计算平台）

11-13/23:28:18|INFO |base_client.py:292|MRPC Train, Client:14, Loss:0.318, Accuracy:0.939
11-13/23:28:36|INFO |base_client.py:292|MRPC Train, Client:3, Loss:0.313, Accuracy:0.970
11-13/23:28:54|INFO |base_client.py:292|MRPC Train, Client:35, Loss:0.324, Accuracy:0.939
11-13/23:29:13|INFO |base_client.py:292|MRPC Train, Client:31, Loss:0.309, Accuracy:0.970
Traceback (most recent call last):
File "/content/drive/MyDrive/workspace/code/FedETuning/main.py", line 20, in
main()
File "/content/drive/MyDrive/workspace/code/FedETuning/main.py", line 16, in main
trainer.train()
File "/content/drive/MyDrive/workspace/code/FedETuning/trainers/FedBaseTrainer.py", line 91, in train
self.server_manger.run()
File "/content/drive/MyDrive/workspace/code/FedETuning/fedlab/core/network_manager.py", line 38, in run
self.main_loop()
File "/content/drive/MyDrive/workspace/code/FedETuning/trainers/BaseServer/base_server.py", line 253, in main_loop
sender_rank, message_code, payload = self._network.recv()
File "/content/drive/MyDrive/workspace/code/FedETuning/fedlab/core/network.py", line 102, in recv
sender_rank, message_code, content = PackageProcessor.recv_package(
File "/content/drive/MyDrive/workspace/code/FedETuning/fedlab/core/communicator/processor.py", line 118, in recv_package
sender_rank, _, slices_size, message_code, data_type = recv_header(
File "/content/drive/MyDrive/workspace/code/FedETuning/fedlab/core/communicator/processor.py", line 96, in recv_header
dist.recv(buffer, src=src)
File "/usr/local/lib/python3.10/dist-packages/torch/distributed/c10d_logger.py", line 47, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/torch/distributed/distributed_c10d.py", line 1632, in recv
work.wait()
RuntimeError: [../third_party/gloo/gloo/transport/tcp/unbound_buffer.cc:81] Timed out waiting 18000000ms for recv operation to complete
sh: 0: getcwd() failed: Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
sh: 0: getcwd() failed: Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
sh: 0: getcwd() failed: Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
sh: 0: getcwd() failed: Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
python3: can't open file 'main.py': [Errno 107] Transport endpoint is not connected
Exception ignored in: <function Pool.del at 0x7cf9930e17e0>
Traceback (most recent call last):
File "/usr/lib/python3.10/multiprocessing/pool.py", line 271, in del
File "/usr/lib/python3.10/multiprocessing/queues.py", line 371, in put
AttributeError: 'NoneType' object has no attribute 'dumps'