本文主要是介绍解决 :nvrtc: error: invalid value for --gpu-architecture (-arch),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
核心:在显卡安装的cuda版本适配的pytorch中,更换pytorch的版本
刚遇到这个错误时,在网上搜索了一下,感谢博主1和博主2的解决方法带给我的启发。
标题服务器cuda是11.3版本,配置其他环境“御用”的pytorch安装语句
conda install pytorch1.12.1 torchvision0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
但就是使用上述语句安装的pytorch版本会和待调试的代码有冲突
更换为pytorch==1.11.0解决问题!
conda install pytorch1.11.0 torchvision0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
上面语句来自pytorch官网
这篇关于解决 :nvrtc: error: invalid value for --gpu-architecture (-arch)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!