yuquanle

icon indicating an email [email protected]

icon company JD AI labs icon location Beijing icon location 微信公众号：AI小白入门

Results 2 comments of yuquanle

我建了一个微信交流群，欢迎相互交流！

麻烦发个新的

ImportError: flash_attn_2_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda9SetDeviceEi

> python 3.10.14, Cuda 12.1, Ubuntu22.04.4 LTS torch==2.3.0, flash-attn==2.5.8 works (2.5.9post1 has the same failure) Thanks. I try python 3.9.19. torch==2.3.0, flash-attn==2.5.8. It works.