search

MooreThreads / torch_musa

torch_musa is an open source repository based on PyTorch, which can make full use of the super computing power of MooreThreads graphics cards.
Other
313 stars 20 forks source link

torch.nn报错 #64

Open 14128l opened 3 hours ago

14128l commented 3 hours ago

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 commented 3 hours ago

驱动和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