/pytorch/aten/src/ATen/native/cuda/Indexing.cu:605: indexSelectSmallIndex: block: [0,0,0], thread: [0,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
时间: 2024-01-30 11:01:53 浏览: 72
这个错误通常是由于在使用 PyTorch 进行 CUDA 索引操作时出现问题引起的。它表明了一个断言失败,即源张量的索引超出了源张量的选择维度大小。这可能是由于索引超出了范围或选择维度的大小设置不正确导致的。
要解决这个问题,你可以检查以下几点:
1. 确保你的索引值在合法范围内。例如,如果你正在尝试访问张量的第10个元素,而张量的大小只有10个元素,那么这个索引就是非法的。
2. 确保选择维度(即你在索引操作中指定的维度)的大小与源张量的相应维度大小匹配。如果选择维度的大小不正确,就会导致断言失败。
3. 检查你的代码逻辑是否正确。可能是你在进行索引操作时出现了错误。
如果以上步骤都没有解决问题,你可以尝试检查你的代码的其他部分,以确定是否有其他地方导致了这个错误。如果问题仍然存在,你可以提供更多的代码和上下文信息,以便我能够更好地帮助你解决问题。
相关问题
C:\actions-runner\_work\pytorch\pytorch\builder\windows\pytorch\aten\src\ATen\native\cuda\Indexing.cu:1146: block: [18,0,0], thread: [0,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
根据您提供的信息,这似乎是一个来自 PyTorch 的 CUDA 异常。该异常通常表示在 CUDA 加速的代码中,某些索引超出了张量的维度范围。
您可以尝试检查您的代码,看看是否有超出张量维度的索引。此外,您可以尝试升级您的 PyTorch 版本,以查看是否可以解决该问题。如果问题仍然存在,您可能需要在 PyTorch 的 GitHub 存储库中报告该问题,以便开发人员进行进一步的调查和解决。
wandb: Currently logged in as: anony-mouse-584351. Use `wandb login --relogin` to force relogin wandb: wandb version 0.15.3 is available! To upgrade, please run: wandb: $ pip install wandb --upgrade wandb: Tracking run with wandb version 0.12.21 wandb: Run data is saved locally in /kaggle/working/yolov7/wandb/run-20230601_125414-1jenk8d0 wandb: Run `wandb offline` to turn off syncing. wandb: Syncing run run12 wandb: ⭐️ View project at https://wandb.ai/anony-mouse-584351/yolov7-tiny?apiKey=323c78e6a061c91ed778f8bd6fc310953f397638 wandb: 🚀 View run at https://wandb.ai/anony-mouse-584351/yolov7-tiny/runs/1jenk8d0?apiKey=323c78e6a061c91ed778f8bd6fc310953f397638 wandb: WARNING Do NOT share these links with anyone. They can be used to claim your runs. /opt/conda/lib/python3.7/site-packages/torch/functional.py:568: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at /usr/local/src/pytorch/aten/src/ATen/native/TensorShape.cpp:2227.) return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined] WARNING: Dataset not found, nonexistent paths: ['/kaggle/working/yolov7/noduleyolov1iyolov7pytorch/valid/images'] Traceback (most recent call last): File "yolov7/train.py", line 616, in <module> train(hyp, opt, device, tb_writer) File "yolov7/train.py", line 97, in train check_dataset(data_dict) # check File "/kaggle/working/yolov7/yolov7/utils/general.py", line 173, in check_dataset raise Exception('Dataset not found.') Exception: Dataset not found. wandb: Waiting for W&B process to finish... (failed 1). Press Control-C to abort syncing. wandb: wandb: Synced run12: https://wandb.ai/anony-mouse-584351/yolov7-tiny/runs/1jenk8d0?apiKey=323c78e6a061c91ed778f8bd6fc310953f397638 wandb: Synced 5 W&B file(s), 0 media file(s), 0 artifact file(s) and 0 other file(s) wandb: Find logs at: ./wandb/run-20230601_125414-1jenk8d0/logs
这段信息看起来像是由Wandb库生成的训练日志。首先,它提示你当前登录的帐户是“anony-mouse-584351”,并提供了升级Wandb库的信息。然后它显示了正在追踪的运行信息,包括运行的本地保存路径和Wandb项目和运行的URL链接。它还提供了警告,防止别人使用这些链接来访问或声明您的运行。最后,它显示了一个异常,指出找不到数据集。可能需要检查数据集路径或确保数据集已正确加载。