我的第一篇博客,记录下我是如何克服被tensorflow环境支配的恐惧,同时感谢网络上各位大神给出的教程帮助。
一、Ubuntu16.04
Ubuntu16.04是之前电脑上就安装好的,这里不做安装介绍,看网上的信息建议大家安装Ubuntu的14.04,16.04或者18.04这三个版本,因为这三个版本的比较稳定,且都有长期的支持维护。
二、显卡驱动的安装
2.1 驱动的下载
这里需要你的显卡是NVIDIA的显卡,驱动下载地址:https://www.geforce.cn/drivers,选择自己机器对应的显卡型号后会出现下边很多个版本的选择,注意这里版本号的选择要与CUDA的版本号相对应[1]
图一
注意这里版本号的选择要与CUDA的版本号相对应[1],如下图二中CUDA的版本号384.81,则驱动版本为384.13,其中.13小于.81,如下图三。(博主在写本篇博客时,已无法在官网找到1080ti的384.13版本。。。而且博主一开始就是按照此规则下载的驱动,因此并不能确定如果不按照此规则是否会对安装结果产生影响。)
图二
图三[1]
2.2 驱动的安装
2.2.1 禁用nouveau驱动
Ubuntu系统在安装时会自带一个nouveau驱动,该驱动是一个开源的支持Nvidia显卡的驱动,我们要进行深度学习的工作时必须要将该驱动更换为NVIDIA的官方驱动。
首先,在终端运行
lsmod | grep nouveau
若有输出,说明nouveau驱