RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

我的PyTorch version 0.4.1,
cudatoolkit 9.0
cudnn 7.6.5
在程序中插入下面这句话,很好地解决了!至于更深入的原因目前不是很清楚,其中解决方案来源于
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

torch.backends.cudnn.benchmark = True

你可能感兴趣的:(pytorch,python,科研)