一般就是因为自己下载的是GPU版本的pytorch,所以需要将变量和model搬到GPU上才可以运行下去,如果不想这样可以用annaconda安装一个CPU版本的pytorch也可以的。
from torch.autograd import Variable
import torch.nn as nn
import torch
conv1 = nn.Conv1d(in_channels=256,out_channels = 100, kernel_size = 2)
input = torch.randn(32, 35, 256)
input = input.permute(0, 2, 1)
print(input.numpy().shape)
input = Variable(input)
device = torch.device('cuda')
input = input.to(device)
conv1 = conv1.to(device)
out = conv1(input)
print(out.size())