Jittor-summary icon indicating copy to clipboard operation
Jittor-summary copied to clipboard

强制summary使用cpu计算

Open cjld opened this issue 3 years ago • 1 comments

https://github.com/liuruiyang98/Jittor-summary/blob/3842c79c4db59b03a5acab1aa8ba23d6085ed53b/jittorsummary/jittorsummary.py#L13

可以在这一行添加一个修饰符@jt.flag_scope(use_cuda=0)来强制使用cpu

cjld avatar Mar 04 '21 05:03 cjld

将jt.int64更换为numpy.int64,现在jittorsummary已支持CUDA版本,通过 device=‘cpu’ 或者 device=‘cuda' 进行控制。

liuruiyang98 avatar Mar 04 '21 06:03 liuruiyang98