cuda使用

　　

import torch
x = torch.randn(2, 3)
x = x.cuda()

print (x)
#最简单的cuda使用,但是感觉好慢啊

posted on 2018-03-13 21:32  张博的博客  阅读(311)  评论(0)  编辑  收藏  举报

导航