BERT版本要求

tensorflow >= 1.11.0 # CPU Version of TensorFlow.
#tensorflow-gpu >= 1.11.0 # GPU version of TensorFlow.

如果選擇 tensorflow_gpu-1.11.0

python版本:2.7, 3.3-3.6
GCC 4.8
Bazel 0.15.0
CUDA9.0 下載地址:https://developer.nvidia.com/cuda-90-download-archive
cuDNN7.1 下載地址:https://developer.nvidia.com/rdp/cudnn-archive

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章