Error while trying to train depth estimation model

[saqib736]

Hello,

I am getting following error:

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

While trying to train the depth estimation model using the following command:

bash tools/dist_train.sh configs/custom/custom_config.py 1

I have one GPU in my system (RTX 4090), can you help me with this issue, i did troubleshooting but couldn't figured it out what is the cause of this issue.