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

[JUSSY21]

运行train.py时发生以下错误

[takaqaq]

你解决这个问题了吗

[prexhu]

You can try another PyTorch version:

# This is not a command can be run directly!
setuptools==59.5
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117

torch+cu117 fixed the GPU architecture error for 4090 in my case😺.