请问使用tensorflow-gpu 什么版本，CUDA和cudnn对应什么版本

macanv / BERT-BiLSTM-CRF-NER

Tensorflow solution of NER task Using BiLSTM-CRF model with Google BERT Fine-tuning And private Server services

https://github.com/macanv/BERT-BiLSMT-CRF-NER

4.69k stars 1.25k forks source link

Closed caseware66 closed 4 years ago

caseware66 commented 4 years ago

目前安装tensorflow 1.13.0 cuda 10.1，cudnn7.6 GPU利用率为0，cpu爆满，模型运行超慢，找不到原因

macanv commented 4 years ago

tensorflow-gpu吧，其他任务也是不会自动到GPU么？

ChChwang commented 4 years ago

哪个版本呢