|
|
|
|
|
|
by mappu
916 days ago
|
|
|
That seems to be the case for me out-of-the-box. $ python3 -m venv .
$ ./bin/pip3 install torch --index-url https://download.pytorch.org/whl/rocm5.6
$ ./bin/python3 -c 'import torch; print([torch.cuda.is_available(), torch.cuda.get_device_name(0)])'
[True, 'AMD Radeon RX 6600 XT']
|
|
|