torch.nn报错

[14128l]

File "/program/multiclassification/classify/yolov8/yolo_dalei.py", line 212, in detect_image
outputs = self.net(images)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 183, in forward
return self.module(*inputs[0], **module_kwargs[0])
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/program/multiclassification/classify/yolov8/nets/yolo.py", line 124, in forward
feat1, feat2, feat3 = self.backbone.forward(x)
File "/program/multiclassification/classify/yolov8/nets/backbone.py", line 126, in forward
x = self.stem(x)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/program/multiclassification/classify/yolov8/nets/backbone.py", line 35, in forward_fuse
return self.act(self.conv(x))
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 460, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 456, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: _Map_base::at

[14128l]

驱动和musa信息

musa_toolkits:
{
"version": "3.1.0",
"git branch": "kuae1.3.0_musa3.1.0",
"git tag": "No tag",
"commit id": "38fe8761d291e32cad642435d9f84c97e4a37bba",
"commit date": "2024-09-05 19:33:56 +0800"
}
mcc:
{
"version": "3.1.0",
"git branch": "kuae1.3.0_musa3.1.0",
"git tag": "No tag",
"commit id": "baf70da0ba9f1a95a844726d8b2c28a1365b886a",
"commit date": "2024-10-08 16:20:58 +0800"
}
muPP:
{
"version": "1.7.0",
"build archs": "21;22",
"git branch": "kuae1.3.0_musa3.1.0_muPP1.7.0",
"git tag": "20241009_develop",
"commit id": "f0e7e1c596943e2fe98e14ff096e3c2f2c877247",
"commit date": "2024-08-23 10:42:21 +0800"
}
mublas:
{
"version": "1.6.0",
"build archs": "21,22",
"git branch": "kuae1.3.0_musa3.1.0_mublas1.6.0",
"git tag": "No tag",
"commit id": "7bd80754907ffe182da297f54737c74f6e4d4e09",
"commit date": "2024-09-30 11:42:09 +0800"
}
mudnn:
{
"version": "2.7.0",
"git branch": "kuae1.3.0_musa3.1.0_mudnn2.7.0",
"git tag": "No tag",
"commit id": "2fa2f81bf0d77fa977ee398f37a3fdfc6d18cfb1",
"commit date": "2024-09-14 16:31:08 +0800"
}
mufft:
{
"version": "1.6.0",
"build archs": "21;22",
"git branch": "kuae1.3.0_musa3.1.0_muFFT1.6.0",
"git tag": "No tag",
"commit id": "a4790e6ac030a68fb033928c54b2d5044d33cf9e",
"commit date": "2024-08-23 10:33:45 +0800"
}
murand:
{
"version": "1.0.0",
"build archs": "21;22",
"git branch": "kuae1.3.0_musa3.1.0_muRand1.0.0",
"git tag": "20241009_develop",
"commit id": "bc218b10bf4ab6454fe03816297478391ee6d503",
"commit date": "2024-05-17 15:37:47 +0800"
}
musify:
{
"version": "1.0.0",
"git branch": "kuae1.3.0_musa3.1.0_musify1.0.0",
"commit id": "d0720f523c8744a882484be081133054ebd9012f",
"commit date": "2024-08-23 11:19:27 +0800"
}
musolver:
{
"version": "1.0.0",
"build archs": "21,22",
"git branch": "kuae1.3.0_musa3.1.0_musolver1.0.0",
"git tag": "No tag",
"commit id": "65e4a393fa32b27068a993d8bf080645dd16f534",
"commit date": "2024-08-29 10:20:23 +0800"
}
musparse:
{
"version": "1.1.0",
"build archs": "21,22",
"git branch": "kuae1.3.0_musa3.1.0_musparse1.1.0",
"git tag": "20240909_develop",
"commit id": "73a84b13534251f7d0ea4d832c9f11839914c59a",
"commit date": "2024-08-23 10:32:58 +0800"
}
musa_runtime:
{
"version": "3.1.0",
"git branch": "kuae1.3.0_musa3.1.0",
"git tag": "No tag",
"commit id": "6ad4f9378e0db4faa38187539e72cc90108ffe9e",
"commit date": "2024-08-26 13:49:17 +0800"
}
driver_dependency:
{
"git branch": "kuae1.3.0_musa3.1.0",
"git tag": "No tag",
"commit id": "db329f8fb8cef9c1c96b01a50688cd3923361973",
"commit date": "2024-10-09 17:43:11 +0800"
}

torch 2.2.0a0+git8ac9b20
torch_musa 1.3.0
torchvision 0.17.2+c1d70fe

---

mthreads-gmi:1.8.2           Driver Version:2.5.0

---

ID Name |PCIe |%GPU Mem
Device Type |Pcie Lane Width |Temp MPC Capable
+-------------------------------------------------------------+
0 MTT S80 |00000000:41:00.0 |0% 10MiB(16384MiB)
Physical |16x(16x) |54C YES

---

Processes:
ID PID Process name GPU Memory
Usage
+-------------------------------------------------------------+
0 1401 /usr/lib/xorg/Xorg 6MiB