torch.set_default_tensor_type('torch.cuda.FloatTensor') 此代码可以使得通过torch定义的对象全部为cuda版本.
【Pytorch】怎么快速的定义cuda版本的变量和网络
最新推荐文章于 2024-05-07 12:56:36 发布
torch.set_default_tensor_type('torch.cuda.FloatTensor') 此代码可以使得通过torch定义的对象全部为cuda版本.