Open jxxtin opened 3 months ago
Hi! Thanks for your interest.
It looks like you installed a wrong torch or cuda version.
Could you please show me your cuda version with nvcc -V
?
Thanks for quick reply @buaacyw
My nvcc version is 11.7 and pytorch is 2.0.1+cu117 as follow
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Jun__8_16:49:14_PDT_2022
Cuda compilation tools, release 11.7, V11.7.99
Build cuda_11.7.r11.7/compiler.31442593_0
import torch
torch.>>> torch.__version__
'2.0.1+cu117'
Hi! I haven't tried flash-attn on CUDA11.7 but I think it should work for 11.7. I suggest you ask about this in the flash-attn repo.
Thank you for the great work. Could you please check the following error that occurred while executing the app.py?