search

rgcottrell / pytorch-human-performance-gec

A PyTorch implementation of "Reaching Human-level Performance in Automatic Grammatical Error Correction: An Empirical Study"
Apache License 2.0
50 stars 19 forks source link

About cuda #1

Closed SongFGH closed 4 years ago

SongFGH commented 5 years ago

Hi, I have a probelm:RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS. And, you write this sentence:If RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS occurs during training, try install pytorch with CUDA 9.2 using conda instead of using default CUDA 9.0. Can you explain this sentence? I usually install pytorch using pip

tianfeichen commented 4 years ago

We used CUDA 9.2 and never tried it on CUDA 9.