DiffusionDet
DiffusionDet copied to clipboard
RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch) ON RTX4090
Excuse me: When i run python train_net.py --num-gpus 1 --config-file configs/diffdet.coco.res50.yaml on 4090,some error occurs.
RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)
Anyone else have account the problem?