bert-pytorch安装记录

根据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