Using user installed CUDA

[davisyoshida]

I'm trying to run some JAX code on a shared cluster, where I can only install CUDA as a user (i.e. I can't symlink my CUDA installation into /usr/local). Is there any way to get around this, or is there no way to change where JAX will look for CUDA?

Edit: I reinstalled CUDA (still as non-root), and things seem to be working now. I'm not sure what exactly fixed it though.