pytorch不能在gpu上运行LSTM

时间:2019-12-27 07:23:51

标签: pytorch lstm

import torch

from torch import nn

class LSTM(nn.module):

     def __init__(self):
          super(LSTM,self).__init__()

         self.feature = nn.LSTM(input_size=28,hidden_size=64,num_layers=1)

         self.out = nn.Linear(64,10)

     def forward(x,self):

          feat,(a,b)= self.feature(x.permute(1,0,2),None)

          return self.out(feat[-1,:,:])

net = LSTM().cuda()

我可以在cpu上打印网络的结构,而anaconda返回 (UnicodeDecodeError:utf-8编解码器无法解码位置67处的字节0xb0:无效的起始字节),如果我在gpu上尝试上网。

还有一件事,我可以在我的cuda上运行Lenet,alexnet和vgg。但这不适用于LSTM。

如何演奏?

0 个答案:

没有答案