DiffusionDet icon indicating copy to clipboard operation
DiffusionDet copied to clipboard

RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch) ON RTX4090

Open RhythmOfTheRain-Byte opened this issue 1 year ago • 0 comments

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?

RhythmOfTheRain-Byte avatar Jan 24 '24 12:01 RhythmOfTheRain-Byte