search

lukecq1231 / nli

Enhanced LTSM for natural language inference
Apache License 2.0
265 stars 70 forks source link

GPU #11

Open farimahjohari opened 5 years ago

farimahjohari commented 5 years ago

If I want to use GPU, I will deal with this problem. image

While Google colab uses Tesla k80, what should I do to get the code running on the GPU?

lukecq1231 commented 5 years ago

I seems that cuDNN doesn't work in Colab. Two ways you can try: 1) make cuDNN available 2) change the command in "train.sh", just remove "optimizer_including=cudnn,"