解决pytorch中“cannot run ‘aten::slow_conv3d_forward’”问题记录

使用pytorch-gpu运行模型，数据为三维贝雷砂岩岩心。其中报错如图

if torch.cuda.is_available():
            
          data = data.cuda()
          data = data.type(torch.FloatTensor)
          target = target.cuda()

其中将模型与data放入gpu代码做出如下更改，报错消失。

if torch.cuda.is_available():
            data = data.to(device)
            target = target.to(device)