CUDA_HOME error? AssertionError: CUDA_HOME does not exist, unable to compile CUDA op(s)

[Mi-Peng]

Hi, I try the example code of vit and I get the error : AssertionError: CUDA_HOME does not exist, unable to compile CUDA op(s) ,located at source code. Did I miss something ? How can i repair the code or the processing?

[Taka152]

Have you installed torch and CUDA correctly, the lightseq can only run torch with CUDA support.

[Mi-Peng]

Have you installed torch and CUDA correctly, the lightseq can only run torch with CUDA support.

Yes. I have installed pytorch 1.11.0+cuda11.3

[Taka152]

Maybe you can try 1.8.0 and 1.10.0, we have tested on these two version.

[SysuJayce]

Same problem with pytorch 2.1.0 + cuda11. Any suggestion?