Traceback (most recent call last): File "E:\yolov5-master\train.py", line 642, in <module> main(opt) File "E:\yolov5-master\train.py", line 531, in main train(opt.hyp, opt, device, callbacks) File "E:\yolov5-master\train.py", line 286, in train for i, (imgs, targets, paths, _) in pbar: # batch ------------------------------------------------------------- File "F:\conda\envs\yolov5\lib\site-packages\tqdm\std.py", line 1178, in __iter__ for obj in iterable: File "E:\yolov5-master\utils\dataloaders.py", line 172, in __iter__ yield next(self.iterator) File "F:\conda\envs\yolov5\lib\site-packages\torch\utils\data\dataloader.py", line 633, in __next__ data = self._next_data() File "F:\conda\envs\yolov5\lib\site-packages\torch\utils\data\dataloader.py", line 1328, in _next_data idx, data = self._get_data() File "F:\conda\envs\yolov5\lib\site-packages\torch\utils\data\dataloader.py", line 1284, in _get_data success, data = self._try_get_data() File "F:\conda\envs\yolov5\lib\site-packages\torch\utils\data\dataloader.py", line 1145, in _try_get_data raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) from e RuntimeError: DataLoader worker (pid(s) 19664, 8920, 6940, 11904, 24688) exited unexpectedly
时间: 2023-07-21 12:51:25 浏览: 72
这个错误是由于使用 DataLoader 时,其中一个 worker 进程异常退出导致的。可以尝试以下方法来解决问题:
1. 检查数据是否有损坏或格式错误。
2. 减少 batch_size 的大小。
3. 减少 num_workers 的数量。
4. 确保 PyTorch 版本和 CUDA 版本兼容。
5. 尝试使用 `pin_memory=False`。
6. 确保 CPU 和 GPU 内存充足。
7. 尝试使用 `multiprocessing.set_start_method('spawn')`,这个设置会在每个 worker 进程中启动一个新的 Python 解释器,可能会解决某些问题。
希望这些方法能够帮助你解决问题。
相关问题
Traceback (most recent call last): File "E:\yolov5-master\train.py", line 642, in <module> main(opt) File "E:\yolov5-master\train.py", line 517, in main device = select_device(opt.device, batch_size=opt.batch_size) File "E:\yolov5-master\utils\torch_utils.py", line 118, in select_device assert torch.cuda.is_available() and torch.cuda.device_count() >= len(device.replace(',', '')), \ AssertionError: Invalid CUDA '--device 3' requested, use '--device cpu' or pass valid CUDA device(s)
根据你提供的错误信息,这是在运行YOLOv5的训练脚本时出现的错误。错误信息显示,你使用了无效的CUDA设备选项"--device 3",建议你检查一下你的CUDA设备是否正确安装并配置好了。你可以尝试使用有效的CUDA设备选项,例如"--device 0"或"--device 1"。如果你只想使用CPU进行训练,则可以使用"--device cpu"选项。
Traceback (most recent call last): File "D:\PyCharm\pytorch_sac-master\train.py", line 152, in <module> @hydra.main(config_path="config/train.yaml", strict=True) TypeError: main() got an unexpected keyword argument 'strict'
这个错误可能是因为你的 hydra 版本过低导致的,建议你升级 hydra 的版本。
你可以使用以下命令来升级 hydra:
```
pip install --upgrade hydra-core
```
如果你使用的是 conda 环境,则可以使用以下命令:
```
conda update hydra-core
```
升级后再次运行程序,看是否能够解决问题。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![whl](https://img-home.csdnimg.cn/images/20210720083646.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![pptx](https://img-home.csdnimg.cn/images/20210720083543.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)