RuntimeError: Detected that PyTorch and torch_butterfly were compiled with different CUDA versions. PyTorch has CUDA version 11.3 and torch_butterfly has CUDA version 12.0.

[abhishektyaagi]

Hi,
I am trying to install torch_butterfly using the setup.py file. After installation, when I import torch_butterly, I get the following error:

RuntimeError: Detected that PyTorch and torch_butterfly were compiled with different CUDA versions. PyTorch has CUDA version 11.3 and torch_butterfly has CUDA version 12.0. Please reinstall the torch_butterfly that matches your PyTorch install

How can I install torch_butterfly with a specific cuda version, considering that I have multiple CUDA versions installed on the server?