【报错解决】RTX4090 nvrtc: error: invalid value for --gpu-architecture (-arch)
在配深度学习环境每次想要训练模型都出现nvrtc: error: invalid value for --gpu-architecture (-arch)这个问题,换了无数个方法都没解决,一开始下载的torch和cuda版本是1.12.0+11.3,后来查了GitHub发现4090框架不支持还是怎么着,最后把torch版本卸掉重新下了1.13+11.6的版本,终于跑通了呜呜太感人了!!
GitHub解决方案链接:nvrtc: error: invalid value for --gpu-architecture (-arch) #87595
下载torch命令:
pip install torch==1.13.0+cu116 torchaudio==0.13.0+cu116 torchvision==0.14.0+cu116 --index-url https://download.pytorch.org/whl/cu116