写在最前面, 因为我们用的tensorflow这个谷歌的AI框架,而且用的是官网的按照保,所以下面的都是按照tensorflow官网的要求和规则来做。其他自己框架或者自己编译的安装的小伙伴们不一定适用。
第一步其实并不是安装安装 NVIDIA cuda, 而是先要选择好操作系统,这个很重要因为现在tensorflow官网只对ubuntu, maxOS, windows的错误进行维护,其他系统的错误就要靠自己囖。
说操作系统重要还有一点就是cuda会依赖不同的操作系统的自带驱动包,然后要根据你的显卡核心版本去下载不同版本的cuda的安装包,不同的tensorflow版本又只能支持对应版本的cuda驱动,如果错了一点可能又要删除重新弄,建议如果是新机器还是用最新的稳定版的系统为好。下面的步骤是针对ubuntu16.0.4和centOS 7的,而且亲测有效(花了两三天时间才装好的)
==========================================真正的开始===========================================
安装 NVIDIA cuda
文档:
总目录:https://developer.nvidia.com/cuda-toolkit-archive
最新版本:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-nouveau
自选版本:https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html
1. 选择 runfile下载 cuda_9.0.176_384.81_linux-run, 在发帖之前tensorflow到了1.9版本,但是官方的安装包都只支持cuda9.0,更高版本的cuda需要自行修改源码去支持了,所以这里选择了9.0的包。在下载之前官网文档上还有一个Pre-installation Actions 的步骤,这个步骤里面的每一小步也是很重要的所以大