【经验】FlashAttention is only supported on CUDA 11.6 and above. Note: make sure nvcc has a supported ve

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

在安装FlashAttention时碰到CUDA版本过低的问题。

Description

lucy66666

opened on Feb 14, 2024

I receive the error below:
RuntimeError: FlashAttention is only supported on CUDA 11.6 and above. Note: make sure nvcc has a supported version by running nvcc -V.
But my pytorch version is 2.2.0+cu118, why is it giving me the error?

RuntimeError: FlashAttention is only supported on CUDA 11 and above #1070

发现nvcc -V 版本出来确实是11.5

没有root权限,没法整体更新

然后发现自己本地目录下其实是有CUDA 12.0 的

解决办法,进入home目录下的用户目录,去修改.bashrc
 

cd /hom

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值