bert 环境搭建之Pytorch&Transformer 安装

本文记录了解决BERT项目中遇到的transformers版本不匹配及安装问题的过程,包括降低transformers版本的方法、解决安装过程中的依赖问题,以及安装PyTorch的具体步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

这两天跑以前的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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值