lukecq1231/nli

GPU

Opened this issue · 1 comments

farimahjohari commented

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

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,"