Skip to content

Enhance docstring for ToDeviced #2864

@yiheng-wang-nv

Description

@yiheng-wang-nv

It seems ToDeviced cannot work when num_workers > 1:

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method

Shall we add the docstring for this transform, thus users can avoid meeting error?

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions