错误

好久没玩torch了，直接装了个最新版本的torch2.0.1+cu117版本， python是3.9

之后在运行代码的时候出现了以下的错误

RuntimeError: GET was unable to find an engine to execute this computation

如果要是cpu跑没有任何报错，但是如果是gpu跑的话就会出现如上错误

class CNNModel(nn.Module):
    def __init__(self, input_size, num_classes):
        super(CNNModel, self).__init__()
        self.conv = nn.Sequential(
            nn.Conv1d(in_channels=input_size, out_channels=32, kernel_size=3, padding=1),
            nn.ReLU(),
            nn.Conv1d(in_channels=32, out_channels=64, kernel_size=3, padding=1),
            nn.ReLU(),
            nn.Conv1d(in_channels=64, out_channels=128, kernel_size=3, padding=1),
            nn.ReLU(),
        )
        self.fc1 = nn.Sequential(
            nn.Linear(128, 64),
            nn.ReLU(),
            nn.Dropout(0.5),
        )
        self.fc2 = nn.Linear(64, num_classes)

    def forward(self, x):
        out = self.conv(x)
        out = torch.flatten(out, start_dim=1)
        out = self.fc1(out)
        out = self.fc2(out)
        return out

x = torch.zeros(64, 128, 1).cuda()
model = CNNModel(input_size=128, num_classes=6)
model.cuda()
model(x)

跑这个代码

解决方案

我之前写的大多都是torch1.x的操作

torch2.0 看文档主要是提升了模型运行速度的能力（针对大模型，新增了torch.compile能力），可以向下兼容，但是我发现很多人在使用过程中都出现了这个问题，具体原因还没有搞清楚，主要出现在conv1d的环节，lstm正常。

解决方案就是把目前这个torch卸载掉，安装以前的版本

我这里使用的依旧是anaconda，我建议直接把虚拟环境删掉，重新建立，因为torch卸载有的时候不干净，安装新的torch会有各种冲突，很麻烦。

假设我门目前使用的虚拟环境时new_env1

# 删掉该环境（谨慎点，如果不方便可以先不删，重新建立新的环境）
conda remove -n new_env1 --all
# 创建新的虚拟环境
conda create -n new_env2 python=3.9

因为torch torchvision cuda以及python版本都是一一有兼容关系的，所以不要乱安装，并且直接安装都时cpu版本，还得再安装对应版本的cuda，更麻烦。

最好这样，直接一个命令搞定

查看：https://pytorch.org/get-started/previous-versions/

亲测有效：conda安装一般都是不好使的，使用pip安装

比如安装torch1.9.0

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

在运行就正常了。

GET was unable to find an engine to execute this computation解决方案

错误

解决方案

猜你喜欢