pytorch报错RuntimeError:error in LoadLibraryA

问题描述

原因一 ：pytorch1.4版本问题

解决办法：

在import torch这里改成下面的代码：

import torch
import ctypes
ctypes.cdll.LoadLibrary('caffe2_nvrtc.dll')
a: torch.Tensor = torch.arange(9).view(3, 3).cuda()
torch.cat([a, a])
torch.stack([a, a])

原因二 ：安全软件误删nvrtc64_101_0.dll文件

解决办法：

恢复误删文件