pytorch如何使用cuda?

pytorch如何使用cuda?
已邀请:
匿名用户

匿名用户

赞同来自:

if torch.cuda.is_available():
    x = x.cuda()
    y = y.cuda()
    x + y
匿名用户

匿名用户

赞同来自:

 直接在需要使用CUDA的张量后加.cuda()即可

要回复问题请先登录注册