【报错解决】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命令:

bash 复制代码
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
相关推荐
AIGC小尼6 分钟前
本地AI漫剧部署|低配8G显存实战部署MiniMax-H3|4步极速采样+低显存优化完整方案(可角色替换/视频魔改)
人工智能·stable diffusion·comfyui·ai漫剧
白猫不黑19 分钟前
Python实现简易Web弱口令爆破与防护方案
python·web安全·计算机·网络安全·黑客·信息安全·渗透测试
宿州派大星42 分钟前
[NLP实战] 基于PyTorch实现N-gram词嵌入模型:输入4个词预测第5个词
人工智能·pytorch·深度学习·nlp
kuuailetianzi1 小时前
Python初识:定位、优势与发展历程
python
guoran_shini1 小时前
LLM微调-训练垂类问答模型
python·lora·sft
qq_199886871 小时前
第8板块·第2节:统一内存的高级特性与性能调优
c++·人工智能·gpu算力·cuda
basketball6161 小时前
Python FastAPI 介绍以及常用方法
python·fastapi·vllm·ai infra
ting94520001 小时前
深度拆解Enter Pro AI原生开发平台:从底层架构到企业级落地技术实践
人工智能·架构·ai-native
lisw051 小时前
人工智能辅助科学的快与慢!
人工智能
9呀1 小时前
vscode如何打开多个codex标签页
人工智能