【解决办法】A6000网络训练时报错nvrtc: error: invalid value for --gpu-architecture (-arch)
【摘要】A6000显卡训练时出现"NVRTC错误：无效的--gpu-architecture参数"，原因是CUDA架构不兼容。该错误表明指定的GPU架构不被当前CUDA工具链支持。解决方案是升级PyTorch环境至匹配版本：通过pip安装torch 1.13.1+cu117、torchvision 0.14.1+cu117和torchaudio 0.13.1，并指定PyTorc
·
报错
A6000深度网络训练时报错nvrtc: error: invalid value for --gpu-architecture (-arch) ，是NVIDIA运行时编译库（NVRTC）报出的错误，通常发生在使用CUDA代码的即时编译（JIT）过程中。该错误表明用户指定的GPU架构（-arch参数）不被当前CUDA工具链支持。
解决办法
参照NVIDIA官方文档，原因是架构不匹配，升级至torch==1.13.0+cu17即可解决
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
更多推荐
4
0- 0
已为社区贡献1条内容
所有评论(0)