bert 环境搭建之Pytorch&Transformer 安装

这两天跑以前的bert项目发现突然跑不了,报错信息如下: 

Step1 transformer 安装

RuntimeError: Failed to import transformers.models.bert.modeling_bert because of the following error (look up to see its traceback):
module 'signal' has no attribute 'SIGKILL'

通过查找信息发现我的transformer 版本太高了 已经4.24了(前几天做了OpenAI的东西,结果把transformer升级了,所以每做一个项目,最好重新建立一个环境),

所以我想把transformer降到3.0。于是直接pip install transformer ==3.00,但是又失败了。失败信息如下:

error: can't find Rust compiler
      
    If you are using an outdated pip version, it is possible a prebuilt wheel is available for this package but pip is not able to install from it. Installing from the wheel would avoid the need for a Rust compiler.
简单看了一下就是没有Rust

所以,我又上官网安装了 Rust (安装 Rust - Rust 程序设计语言 (rust-lang.org)),以为这次肯定成功,然后又pip install transformer ==3.00,结果又报错:

this version of Cargo is older than the `2021` edition, and only supports `2。

到这里我想了一下,以前安装transformer 没有那么麻烦,至少肯定没有用到Rust, 所以我猜测我的python=3.9也是太高了,所以又重新建立python=3.7的环境。

直接pip install transformer ==3.00 终于成功了!

Step2 Pytorch 安装

理论上很简单直接,官方Start Locally | PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

复制这段文字安装,但是发现实在太慢了,后来又用网上介绍的用清华镜像,但是这样安装速度也不是很快,而且下载Pytorch 没有GPU ,具体原因为啥不知道,理论上是可以的。

所以最后直接找到 https://download.pytorch.org/whl/torch/

下载了适合的版本,这样下载很快们都是10M/S

bert 环境搭建之Pytorch&Transformer 安装_第1张图片

 写这篇文章主要问题就是记住这个下载的网址

你可能感兴趣的:(transformer,bert,pytorch)