Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should b

2022-05-12 08:35:31 浏览数 (1)

问题描述Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
在使用pytorch训练经典的MNIST数据集时,运行时,出现了以下的问题:
问题原因: 错误内容大概就是指输入类型是CPU（torch.FloatTensor），而参数类型是GPU（torch.cuda.FloatTensor）报错内容是:输入的是CPU类型的(torch.FloatTensor),然而输出的内容是GPU类型的,同时它提示,应该保持一直的数据类型
解决错误: 首先检查我们是不是正确的使用了CUDA:

1.下面是正确的使用CUDA的方法:

代码语言：txt复制

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

2.而我之前在使用CUDA进行加速时,是这样写的:

代码语言：txt复制

 if torch.cuda.is_available():
	model.cuda()

显然这样写是错误的,而应该采用第一种方法

0 人点赞