We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Jittor-summary/jittorsummary/jittorsummary.py
Line 13 in 3842c79
可以在这一行添加一个修饰符@jt.flag_scope(use_cuda=0)来强制使用cpu
@jt.flag_scope(use_cuda=0)
The text was updated successfully, but these errors were encountered:
[#1] Adding support for CUDA
f686443
将jt.int64更换为numpy.int64,现在jittorsummary已支持CUDA版本,通过 device=‘cpu’ 或者 device=‘cuda' 进行控制。
Sorry, something went wrong.
No branches or pull requests
Jittor-summary/jittorsummary/jittorsummary.py
Line 13 in 3842c79
可以在这一行添加一个修饰符
@jt.flag_scope(use_cuda=0)
来强制使用cpuThe text was updated successfully, but these errors were encountered: