Transformer安装过程中报没有apex安装包的问题

解决CUDA计算力配置错误安装NVIDIA apex库
在尝试安装NVIDIA apex库时遇到错误,安装过程中的python setup.py install命令报错。问题在于CUDA计算力值未正确设置。通过在命令行中输入export TORCH_CUDA_ARCH_LIST=6.0(可替换为任意计算力值)并重新执行安装命令,成功解决了安装问题,最终完成apex库的安装。这个小插曲耗费了不少时间,但最终还是令人欣慰地解决了。
部署运行你感兴趣的模型镜像

解决方案:

git clone https://github.com/NVIDIA/apex

cd apex

python setup.py install

在使用python setup.py install 的过程中会报错, 错误截图

解决方案:

需要在命令行执行 export TORCH_CUDA_ARCH_LIST="6.0" 随便一下计算力值输入即可.然后在执行python setup.py install 即可以安装成功.

啊.....总算安装成功, 在这里卡了好长, 心酸...... 大腿都拍肿了!!!

您可能感兴趣的与本文相关的镜像

Python3.10

Python3.10

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值