分类：tensor定义

2020
10-08

pytorch cuda上tensor的定义 以及减少cpu的操作详解

cuda上tensor的定义a=torch.ones(1000,1000,3).cuda()某一gpu上定义cuda1=torch.device('cuda:1')b=torch.randn((1000,1000,1000),device=cuda1)删除某一变量dela在cpu定义tensor然后转到gputorch.zeros().cuda()直接在gpu上定义，这样就减少了cpu的损耗torch.cuda.FloatTensor(batch_size,self.hidden_dim,self.height,self.width).fill_(0)补充知识：pytorchcuda.FloatTensor->FloatTensor错误类型：RuntimeError:In... 继续阅读 >