windows环境下cuda报错:nvcc fatal : Unsupported gpu architecture ‘compute_86‘

向系统环境变量中添加一项:TORCH_CUDA_ARCH_LIST,值为7.5。在这里插入图片描述
重启pycharm或者cmd,再进行相关操作即可。

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐