通常不建议在多进程加载中返回 CUDA 张量,因为在使用 CUDA 和在并行处理中共享 CUDA 张量时存在很多微妙之处(请参见[在并行处理中的 CUDA](notes/multiprocessing.html#multiprocessing-cuda-note))。 相反,我们建议使用[自动内存固定](#memory-pinning)(即,设置`pin_memory=True`),该功能可以将数据快速传输到支持 CUDA 的 GPU。
通常不建议在多进程加载中返回 CUDA 张量,因为在使用 CUDA 和并行处理共享 CUDA 张量时,存在很多微妙之处(请参见[在并行处理中的 CUDA](notes/multiprocessing.html#multiprocessing-cuda-note))。相反,我们建议使用[自动内存固定](#memory-pinning)(即,设置`pin_memory=True`),该功能可以将数据快速传输到支持CUDA的GPU中。