mythzZzZ

3090/a100 operating environment

3

3090/a100 operating environment Driver Version: 495.29.05 CUDA Version: 11.5 conda create -n querydet python=3.7 -y source activate querydet pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html python -m pip install detectron2...

hello，What are the parameters of RetinaNet and QueryDetin the paper

请问这是什么原因造成的

Using the train_aux.py training script, single gpu training fails, but multiple gpu training succeeds.

1

Traceback (most recent call last): File "/opt/conda/lib/python3.10/site-packages/torch/cuda/__init__.py", line 306, in _lazy_init queued_call() File "/opt/conda/lib/python3.10/site-packages/torch/cuda/__init__.py", line 174, in _check_capability capability = get_device_capability(d) File "/opt/conda/lib/python3.10/site-packages/torch/cuda/__init__.py", line 430, in get_device_capability prop = get_device_properties(device)...