Yes, you need to install the right version or Cupy hangs forever when installing via pip:

    pip install cupy-cuda12x