修复sm_86错误

要使用正确的Docker镜像

修复sm_86错误

该错误信息为

nvcc fatal   : Value 'sm_86' is not defined for option 'gpu-architecture'

这个问题是在使用TensorFlow的官方Docker镜像时出现。

解决方案:使用Nvidia官方镜像。

TensorFlow Release Notes :: NVIDIA Deep Learning Frameworks Documentation

转发至

微信扫一扫分享

WeChat QR Code

有AI相关的想法,但不知道怎么落地?

我是国冰,独立AI开发者。十多年技术一线——半导体显示、智慧城市到自动驾驶;现在一个人+AI,接AI落地项目,从梳理需求到系统落地,陪你走通。

微信聊聊 微信二维码 看我能做什么