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