Bert-chinese使用GPU运算

Bert-chinese:
基于Tensorflow 1.15.0,需要安装所有CUDA 10.0 libraries:

sudo apt-get install cuda-libraries-10-0


使用以下命令确认GPU是否可用

In [1]: import tensorflow as tf                                                 
In [2]: tf.test.is_gpu_available()

你可能感兴趣的:(机器学习)