jayleicn / moment_detr

[NeurIPS 2021] Moment-DETR code and QVHighlights dataset
https://arxiv.org/abs/2107.09609
MIT License
259 stars 44 forks source link

Can't Run without GPU #48

Open Ccivam opened 9 months ago

Ccivam commented 9 months ago

Traceback (most recent call last): File "moment_detr/train.py", line 255, in best_ckpt_path, eval_split_name, eval_path, debug = start_training() File "moment_detr/train.py", line 246, in start_training model, criterion, optimizer, lr_scheduler = setup_model(opt) File "/home/ciivam/IE643_PROJECT/moment_detr/moment_detr/inference.py", line 195, in setup_model model, criterion = build_model(opt) File "/home/ciivam/IE643_PROJECT/moment_detr/moment_detr/model.py", line 445, in build_model criterion.to(device) File "/home/ciivam/anaconda3/envs/moment_detr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 989, in to return self._apply(convert) File "/home/ciivam/anaconda3/envs/moment_detr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 688, in _apply self._buffers[key] = fn(buf) File "/home/ciivam/anaconda3/envs/moment_detr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 987, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) File "/home/ciivam/anaconda3/envs/moment_detr/lib/python3.7/site-packages/torch/cuda/init.py", line 229, in _lazy_init torch._C._cuda_init() RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

Getting this while doing bash moment_detr/scripts/train.sh.Cant we run it without gpu I know it would take a lot of time