You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
你好,我按照您给出的环境配置本地环境,模型运行到
if torch.cuda.is_available():
model = self.model.to(DEVICE)时,
报错RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM,将模型放在CPU上则没有问题,想请问一下您之前遇到过类似的吗?
The text was updated successfully, but these errors were encountered:
你好,我按照您给出的环境配置本地环境,模型运行到
if torch.cuda.is_available():
model = self.model.to(DEVICE)时,
报错RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM,将模型放在CPU上则没有问题,想请问一下您之前遇到过类似的吗?
The text was updated successfully, but these errors were encountered: