根据Github https://github.com/codertimo/BERT-pytorch 中的readme执行
pip3 install bert-pytorch
结果报错:Couldn't find a version that satisfies the requirement torch>=0.4.0
返回conda list发现torch的版本是0.3.1低于0.4.0, 网上搜了一下发现windows上已支持(参考文章1).
因此先开始升级torch(参考文章2).
在cmd中输入nvcc --version查到cuda的版本,再根据cuda版本号和python版本号去下载对应的torch,
选择对应cuda9.0,python3.6的版本后输入如下命令即可开始升级torch:
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl
安装完成后再输入一遍上述命令则显示已安装成功torch0.4.0,如下图显示:
再次安装bert-pytorch:
输入命令 pip3 install bert-pytorch,这次没有报错,返回conda list显示bert-pytorch已安装成功。
后记:
参考文章4 huggingface版本的bert-pytorch需要升级torch到0.4.1/1.0.0的版本,安装命令为
pip install pytorch-pretrained-bert
本文是按照参考文章5 codertimo版本的bert-pytorch安装的升级到0.4.0版本即可。
参考文章:
1.https://www.analyticsvidhya.com/blog/2018/04/pytorch-0-4-0-windows-support-released/
2.https://blog.csdn.net/sunqiande88/article/details/80085569
3.https://github.com/google-research/bert
4.https://github.com/huggingface/pytorch-pretrained-BERT
5.https://github.com/codertimo/BERT-pytorch
6.https://zhuanlan.zhihu.com/p/50773178