Cannot import on machine with ROCm CPU + CUDA GPU

[reikdas]

>>> import scorch

gets stuck at

No ROCm runtime is found, using ROCM_HOME='/opt/rocm'

However,

>>> import torch
>>> torch.cuda.is_available()
True

works fine.

My CPU is AMD:

Vendor ID:                AuthenticAMD
  Model name:             AMD Ryzen Threadripper PRO 7965WX 24-Cores

And my GPU is:

Fri Sep 27 10:50:07 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.90.07              Driver Version: 550.90.07      CUDA Version: 12.4     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA RTX 6000 Ada Gene...    Off |   00000000:01:00.0 Off |                  Off |
| 30%   36C    P8             24W /  300W |      69MiB /  49140MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
|   1  NVIDIA RTX 6000 Ada Gene...    Off |   00000000:C1:00.0 Off |                  Off |
| 30%   35C    P8             23W /  300W |      12MiB /  49140MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

I suspect that maybe some dependency is seeing that ROCm is available and is getting confused about CUDA and ROCm?