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()
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()