Ubuntu1604+Nvida驱动+CUDA9.0+cuDNN7.0+tensorflow-gpu1.8安装过程

1.Ubuntu安装NVIDIA驱动

Ubuntu安装NVIDA驱动可以直接在软件和更新里安装,也可以自行到NVIDA官网下载合适的驱动安装,驱动不一定要安装最新的,但越是高端的显卡越是要安装稳定的版本。这里我选择的是Ubuntu自带的软件更新里的驱动384.130,够省心。多显卡的只需要装一次,这个与安装驱动个数无关,安装好后执行nvidia-smi会显示出显卡的驱动版本跟其他信息。        

2.安装CUDA9.0

安装cuda切莫追求最新,因为cuda跟cudnn之间的版本兼容性有关系。以下为目前所支持的一些版本搭配:
这里选择cuda9.0和cudnn7.0是因为cudnn可以直接有deb安装。CUDA下载地址,直接下载runfile(local)格式。
执行:sudo ./cuda_9.0.176_384.81_linux.run开始安装
安装过程中除了在选择nvidia driver一步中选no,因为在安装cuda之前已经安装好驱动,其他可以一路yes,最后等待安装成功,最后执行nvcc -V会显示cuda的版本信息。
        

3.安装cuDNN7.0

cuDNN建议下载7.0适配cuda9.0的deb版本,

        

注意有3个的都要下载下来,最后安装顺序也是这3个从上往下安装,因为相互之间有依赖关系。安装好后暂时不需要测试,一般版本不排斥都是有效的。

4.安装tensorflow-gpu1.8.0

安装tensorflow有多种方式,我是选择在anaconda里装的,具体看另一篇使用 anaconda 进行安装tensorflow_gpu

安装好后进行测试,有输入能输出一般没问题:



猜你喜欢

转载自blog.csdn.net/fengliang4616/article/details/81055279