ROCm

Need python-pytorch-rocm. Run with:

~/git/eth/eth-exercises/hodl (main*) » HSA_OVERRIDE_GFX_VERSION=11.0.0 python                                                                                                                                                   thomas@zenbook
Python 3.13.2 (main, Feb  5 2025, 08:05:21) [GCC 14.2.1 20250128] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.rand(10).to('cuda')
tensor([0.6966, 0.0632, 0.5696, 0.9274, 0.5854, 0.1239, 0.0903, 0.2909, 0.3198,
        0.1327], device='cuda:0')
>>>