Importing RRef, rpc_async, remote from RPC

distributed distributed-rpc
1

Hey guys

I just started using RPC yesterday for SEED_RL and I have a problem importing RRef, rpc_async, remote from torch.distributed.rpc

if I run
from torch.distributed.rpc import RRef, rpc_async, remote

It keeps saying, "cannot import name ‘RRef’ or ‘rpc_async’ or ‘remote’ from ‘torch.distributed.rpc’

I tried it on Windows and also WSL (Ubuntu)

2

Hey
@ptrblck
Can you make some suggestions, please?

Thank you

3

I cannot reproduce the issue on Linux:

from torch.distributed.rpc import RRef, rpc_async, remote
RRef
# torch.distributed.rpc.api.RRef
rpc_async
# <function torch.distributed.rpc.api.rpc_async(to, func, args=None, kwargs=None, timeout=-1.0)>
remote
# <function torch.distributed.rpc.api.remote(to, func, args=None, kwargs=None, timeout=-1.0)>

using 2.4.0.dev20240423+cu121.