写在最前面, 因为我们用的tensorflow这个谷歌的AI框架,而且用的是官网的按照保,所以下面的都是按照tensorflow官网的要求和规则来做。其他自己框架或者自己编译的安装的小伙伴们不一定适用。
第一步其实并不是安装安装 NVIDIA cuda, 而是先要选择好操作系统,这个很重要因为现在tensorflow官网只对ubuntu, maxOS, windows的错误进行维护,其他系统的错误就要靠自己囖。
说操作系统重要还有一点就是cuda会依赖不同的操作系统的自带驱动包,然后要根据你的显卡核心版本去下载不同版本的cuda的安装包,不同的tensorflow版本又只能支持对应版本的cuda驱动,如果错了一点可能又要删除重新弄,建议如果是新机器还是用最新的稳定版的系统为好。下面的步骤是针对ubuntu16.0.4和centOS 7的,而且亲测有效(花了两三天时间才装好的)
==========================================真正的开始===========================================
安装 NVIDIA cuda
文档:
总目录:http