[Docs] torch

[Radiantzy]

Branch

master branch https://mmrotate.readthedocs.io/en/latest/

📚 The doc issue

Traceback (most recent call last):
File "tools/train.py", line 128, in
main()
File "tools/train.py", line 124, in main
runner.train()
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\runner.py", line 1777, in train
model = self.train_loop.run() # type: ignore
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 98, in run
self.run_epoch()
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 115, in run_epoch
self.run_iter(idx, data_batch)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 131, in run_iter
outputs = self.runner.model.train_step(
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\model\base_model\base_model.py", line 114, in train_step
losses = self._run_forward(data, mode='loss') # type: ignore
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\model\base_model\base_model.py", line 361, in _run_forward
results = self(**data, mode=mode)
File "F:\anaconda3\envs\orformer\lib\site-packages\torch\nn\modules\module.py", line 1102, in call_impl
return forward_call(*input, **kwargs)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmdet\models\detectors\base.py", line 92, in forward
return self.loss(inputs, data_samples)
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_rcnn.py", line 40, in loss
self.rpn_head.loss_and_predict(
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 242, in loss_and_predict
self.get_inputs(main_results, aux_results, img_metas=img_metas)
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 375, in get_inputs
dqs_all_cls_scores, dqs_all_bbox_preds, dqs_query_ids = self.dqs(
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 468, in dqs
det_bboxes, keep_idxs = batched_nms(mlvl_bboxes,
File "F:\anaconda3\envs\orformer\lib\site-packages\mmcv\ops\nms.py", line 303, in batched_nms
dets, keep = nms_op(boxes_for_nms, scores, **nms_cfg)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmcv\ops\nms.py", line 444, in nms_rotated
keep_inds = ext_module.nms_rotated(dets_wl, scores, order, dets_sorted,
RuntimeError: CUDA error: no kernel image is available for execution on the device
Traceback (most recent call last):
File "tools/train.py", line 128, in
main()
File "tools/train.py", line 124, in main
runner.train()
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\runner.py", line 1777, in train
model = self.train_loop.run() # type: ignore
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 98, in run
self.run_epoch()
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 115, in run_epoch
self.run_iter(idx, data_batch)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\runner\loops.py", line 131, in run_iter
outputs = self.runner.model.train_step(
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\model\base_model\base_model.py", line 114, in train_step
losses = self._run_forward(data, mode='loss') # type: ignore
File "F:\anaconda3\envs\orformer\lib\site-packages\mmengine\model\base_model\base_model.py", line 361, in _run_forward
results = self(**data, mode=mode)
File "F:\anaconda3\envs\orformer\lib\site-packages\torch\nn\modules\module.py", line 1102, in call_impl
return forward_call(*input, **kwargs)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmdet\models\detectors\base.py", line 92, in forward
return self.loss(inputs, data_samples)
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_rcnn.py", line 40, in loss
self.rpn_head.loss_and_predict(
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 242, in loss_and_predict
self.get_inputs(main_results, aux_results, img_metas=img_metas)
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 375, in get_inputs
dqs_all_cls_scores, dqs_all_bbox_preds, dqs_query_ids = self.dqs(
File "H:\OrientedFormer\tools..\projects\OrientedFormer\orientedformer\oriented_ddq_fcn.py", line 468, in dqs
det_bboxes, keep_idxs = batched_nms(mlvl_bboxes,
File "F:\anaconda3\envs\orformer\lib\site-packages\mmcv\ops\nms.py", line 303, in batched_nms
dets, keep = nms_op(boxes_for_nms, scores, **nms_cfg)
File "F:\anaconda3\envs\orformer\lib\site-packages\mmcv\ops\nms.py", line 444, in nms_rotated
keep_inds = ext_module.nms_rotated(dets_wl, scores, order, dets_sorted,
RuntimeError: CUDA error: no kernel image is available for execution on the device

Suggest a potential alternative/fix

No response

[Radiantzy]

Hello, I have tried multiple versions of Torch but all of them show problems. Could you please provide me with your Torch, Torch version, and TorchAudio version?

[wokaikaixinxin]

Please refer ai4rs !
请参考 ai4rs !

[wokaikaixinxin]

or use the old version

conda create --name openmmlab python=3.8 -y
conda activate openmmlab

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch

pip install -U openmim -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install mmengine -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install "mmcv>2.0.0rc4, <2.1.0" -i https://pypi.tuna.tsinghua.edu.cn/simple

mim install 'mmdet>3.0.0rc6, <3.2.0' -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install -v -e . -i https://pypi.tuna.tsinghua.edu.cn/simple