CUDA_HOME not automatically set

I get this error despite running the cuda toolkit step:
```
  OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
```

https://github.com/rsxdalv/flash-attention/actions/runs/11836964853/job/32982975377#step:6:41