Megvii-BaseDetection / Megvii-BaseDetection/YOLOX
RuntimeError: CUDA out of memory. Tried to allocate 4.53 GiB
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 10.6k
- Forks
- 2.5k
- PR merge metrics
- No merged PRs in 30d
Description
python tools/train.py -f exps/example/custom/yolox_s.py -d 1 -b 1 --fp16 -o -c F:\YOLOX\yolox_s.pth
RuntimeError: CUDA out of memory. Tried to allocate 4.53 GiB (GPU 0; 6.00 GiB total capacity; 39.04 MiB already allocated; 4.45 GiB free; 64.00 MiB reserved in total by PyTorch)
（在yolov5和paddle下都可以训练，环境没问题）
显卡 2060 6GB显存
CUDA Version: 11.2
数据集：coco128 （官方演示数据集）
是不是我还有配置文件没有调，还是只能换3060这些显卡？
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the reported command in tools/train.py and the exps/example/custom/yolox_s.py configuration, then reproduce the CUDA out-of-memory error on the stated 6 GB GPU with coco128. Check the training configuration and memory-related options against the documented examples. Done means identifying a project-supported configuration that runs successfully or documenting the confirmed hardware requirement and limitation.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 30/100