/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 21:01:45 浏览: 32
这个错误通常在使用 PyTorch 进行索引操作时出现。它表示在 CUDA 设备上执行索引选择操作时,源索引超过了源张量的维度大小。
要解决这个问题,你可以尝试以下几种方法:
1. 检查你的源张量和索引张量的维度是否匹配。确保索引张量的值在源张量的合法范围内。
2. 确保你的源张量和索引张量都在 CUDA 设备上。如果其中一个张量在 CPU 上而另一个在 CUDA 设备上,可能会导致此错误。
3. 如果你在进行并行操作,例如使用多个线程或多个 GPU,确保每个线程或 GPU 访问的数据都是独立的。避免多个线程或 GPU 访问相同的数据位置。
如果问题仍然存在,请提供更多的代码和上下文信息,以便我能够更好地帮助你解决问题。
相关问题
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链接。它还提供了警告,防止别人使用这些链接来访问或声明您的运行。最后,它显示了一个异常,指出找不到数据集。可能需要检查数据集路径或确保数据集已正确加载。