LeMei/UniMSE

配置环境问题

jiayoujiayoujiayoua opened this issue · 0 comments

你好,我按照您给出的环境配置本地环境,模型运行到
if torch.cuda.is_available():
model = self.model.to(DEVICE)时,
报错RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM,将模型放在CPU上则没有问题,想请问一下您之前遇到过类似的吗?