How to enable CudaGdrChannel registration in tensorpipeAgent when using pytorch's rpc

pytorch / tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

Other

249 stars 75 forks source link

Closed eedalong closed 2 years ago

eedalong commented 2 years ago

Can we just enable it by define some environment variables or we need to recompile pytorch? Thx!

lw commented 2 years ago

It's always enabled by default

eedalong commented 2 years ago

That's great! Thx!