Error when running the training demo

[Ramseyous0109]

Error occurred when running the training demo in README.md:

Traceback (most recent call last):
  File "/home/mt/buaa/try_musa/classification/train_written_network.py", line 64, in <module>
    loss = criterion(outputs, labels.to(device))
  File "/home/mt/buaa/anaconda3/envs/try_musa/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/mt/buaa/anaconda3/envs/try_musa/lib/python3.10/site-packages/torch/nn/modules/loss.py", line 1174, in forward
    return F.cross_entropy(input, target, weight=self.weight,
  File "/home/mt/buaa/anaconda3/envs/try_musa/lib/python3.10/site-packages/torch/nn/functional.py", line 3029, in cross_entropy
    return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: MUSA error: invalid device function

The installation of MUSA and torch-musa is successful, versions listed:

• System: Ubuntu 20.04
• GPU: S80
• MUSA-SDK: rc2.0.0
• MUSA_Toolkit: rc2.0.0
• muDNN: rc2.4.0
• torch_musa-1.1.0-cp310

[hanhaowen-mt]

It seems that the driver is not installed properly. You can try to use the developer version of Docker