pytorch / examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Home Page:https://pytorch.org/examples

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

parameter server example not running in ubuntu

gagandeep987123 opened this issue · comments

I am running the server and worker in separate sessions but execution stalls in the worker thread at line 175 while running
for i, (data, target) in enumerate(train_loader):
However, it is running fine on Mac.
The issue only happens in ubuntu.
Can you please help

@gagandeep987123 thanks for reporting the issue! It seems to be deadlock caused by "fork" start method(default in Linux). In Mac, the default is "spawn" so it works fine. You could try to add "mp.set_start_method("spawn")" before any process starts.

@gagandeep987123 Please feel free to reopen this issue if it's not resolved by #1025