Слияние кода завершено, страница обновится автоматически
Успешно скомпилировал pytorch и torch_musa из исходного кода, следуя инструкциям в README файла репозитория torch_musa. Однако при выполнении следующего кода:
import torch
import torch_musa
a = torch.tensor([1.2, 2.3], dtype=torch.float32, device='musa')
возникает ошибка:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
RuntimeError: MUSA error: no CUDA-capable device is detected
Проверка API torch.musa.is_available()
также вернула значение False
.
Мои системные настройки следующие:
ОС: Ubuntu 20.04
Python: v3.10 в Conda
PyTorch: v2.0.0
Torch_musa: v1.1.0+3a86c94
Устройство GPU: MTT S4000
Версия драйвера: 2.7.0