Vulkan backend on Linux

[faveoled]

Does anyone have success with using Vulkan acceleration on Linux?
I'm getting the following error on my Ubuntu 22.04.3 and AMD GPU:

user@hp-laptop:~/Music$ DRI_PRIME=1 whisper --model tiny.en --language en --device vulkan file.mp3
/usr/lib/python3/dist-packages/scipy/__init__.py:146: UserWarning: A NumPy version >=1.17.3 and <1.25.0 is required for this version of SciPy (detected version 1.25.2
  warnings.warn(f"A NumPy version >={np_minversion} and <{np_maxversion}"
Traceback (most recent call last):
  File "/home/user/.local/bin/whisper", line 8, in <module>
    sys.exit(cli())
  File "/home/user/.local/lib/python3.10/site-packages/whisper/transcribe.py", line 444, in cli
    model = load_model(model_name, device=device, download_root=model_dir)
  File "/home/user/.local/lib/python3.10/site-packages/whisper/__init__.py", line 144, in load_model
    checkpoint = torch.load(fp, map_location=device)
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 809, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1172, in _load
    result = unpickler.load()
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1142, in persistent_load
    typed_storage = load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1116, in load_tensor
    wrap_storage=restore_location(storage, location),
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1083, in restore_location
    return default_restore_location(storage, map_location)
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 220, in default_restore_location
    raise RuntimeError("don't know how to restore data location of "
RuntimeError: don't know how to restore data location of torch.storage.UntypedStorage (tagged with vulkan)

[phineas-pta]

use cpu

if u intall torch with rocm properly u can use "cuda" (even if it's amd gpu) coz that's torch device

[faveoled]

Doesn't seem like rocm supports my GPU sadly