RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

[JaneHobi]

CUDA_VISIBLE_DEVICES=0,1 python train.py --name=YOUR_EXPNAME --q_update --q_attn --basicblock_trans --bfs_clf 报错： ... loading data/dev_graph.pkl.gz Traceback (most recent call last): File "train.py", line 215 in model.cuda(model_gpu) File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 258, in cuda return self._apply(lambda t: t.cuda(device)) File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 185, in _apply module._apply(fn) File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 185, in _apply module._apply(fn) File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 185, in _apply module._apply(fn) [Previous line repeated 1 more time] File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 112, in _apply self.flatten_parameters() File "/root/miniconda3/envs/myconda/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 105, in flatten_parameters self.batch_first, bool(self.bidirection)) RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

我的环境：python 3.6，pytorch 0.4.1，cuda 9.2, cudnn 7.6.5

[saravananpsg]

Hi, Even I have faced the same error when I try to run the model with same config.

cuDNN version: /usr/local/cuda-10.0/lib64/libcudnn.so.7.6.0

Versions of relevant libraries: [pip3] pytorch-pretrained-bert==0.6.2 [pip3] torch==0.4.1.post2 [conda] cuda90 1.0 h6433d27_0 pytorch [conda] pytorch 0.4.1 py37_py36_py35_py27__9.0.176_7.1.2_2 pytorch [conda] pytorch-pretrained-bert 0.6.2 pypi_0 pypi