背景
參照博客:https://blog.csdn.net/macanv/article/details/85684284
參照代碼:https://github.com/macanv/BERT-BiLSTM-CRF-NER
環(huán)境安裝
上述博客已經詳細說明了安裝方法,此處記錄自己安裝環(huán)境中遇到的一些問題及解決方法:
1.執(zhí)行安裝命令pip install bert-base==0.0.7 -i https://pypi.python.org/simple
時另患,一直失敗缨该,報連不上pypi.python.org募舟,看了下機器是可以ping通該域名的年叮。
后來就百度了一下锉试,換了一個源:http://pypi.douban.com/simple
具體操作如下:新建~/pip/pip.conf
文件瘸洛,寫入
[global]
index-url = http://pypi.douban.com/simple
trusted-host = pypi.douban.com
disable-pip-version-check = true
timeout = 120
然后執(zhí)行命令pip install bert-base==0.0.7 -i http://pypi.douban.com/simple
即可安裝成功
2.安裝tensorflow揍移,總是報錯pip._vendor.requests.packages.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='pypi.tuna.tsinghua.edu.cn', port=443): Read timed out.
,主要是網(wǎng)絡問題反肋,解決方案無他那伐,多試試就好了(或者也可以下載好后,直接安裝)
3.遇到cuda版本與tensorflow版本不一致問題石蔗,后來經過查詢罕邀,tensorflow1.12.0 可以安裝在 cuda 9上面附查詢cuda版本命令:cat /usr/local/cuda/version.txt
4.安裝的tensorflow-gpu版本,運行出錯
4.1failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED]
問題养距,經查發(fā)現(xiàn)是因為顯存不夠诉探,后來過了一段時間,再試了一下棍厌,就運行起來了(大家共享的機器);
4.2.OP_REQUIRES failed at assign_op.h:117 : Resource exhausted: OOM when allocating tensor with shape[3072,768] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
參考 引用2肾胯,將batch_size調小
5.在出現(xiàn)gpu版本的問題后,嘗試了tensorflow cpu版本耘纱,運行沒有問題敬肚,但是太慢
引用
1.https://www.cnblogs.com/cuzz/p/8192663.html
2.https://blog.csdn.net/Will_Ye/article/details/89878588