多进程队列 CUDA张量共享 - pytorch吧 - RAT

多进程队列 CUDA张量共享

level 2

thivcd 楼主

CUDA张量能否放到multiprocessing.Queue中？
这里测试时报错。
有没有大佬指点一下。

2022年09月08日 09点09分 1

1