Ubuntu16.04+gtx1070显卡+安装TensorFlowGPU版
首先我先推荐大家去看这2篇我的文章,第一篇讲,怎么制作Ubuntu16的U盘安装,里面没讲怎么安装Ubuntu,我们用的最简单的方法,直接没有保存电脑上面的数据,盘也懒得分了,网上有教程,这里就不细讲了。
大概流程是,开机显示屏提示按F12 你不停的按,进入装机界面,选第5个:USB-uefi .....
接着选第2个,直接按Ubuntu。后面忘了,记得有个选择安装模式,我选第一个,就省掉了分盘操作。
https://mp.youkuaiyun.com/postedit/79722615
第二篇讲了,针对我的1070显卡,怎么选择合适的驱动,怎么安装。可能一般的用户的话安装好了Ubuntu就不用管驱动问题了,但是我们要用来搞学习的(Ubuntu自带显卡不行,TensorFlow必须要用英伟达的显卡+驱动),必须找到对1070最兼容的驱动。所以才有了下面这篇文章。
https://mp.youkuaiyun.com/postedit/79746624
先给大家讲一下我这次的各个版本吧:
系统:Ubuntu16.04
显卡:gtx170
显卡驱动:NVIDIA-Linux-x86_64-390.42.run===英伟达官网输入系统版本,显卡版本,自动匹配下载的
cuda:cuda_9.0.176_384.81_linux.run===因为tf1.6支持9.0,所以下载这个版本cuda
cudnn:cudnn-9.0-linux-x64-v7.tgz ====cudnn给cuda加速的,自然版本要保持一致
anaconda:Anaconda3-4.4.0-Linux-x86_64.sh ===自己网上找的,好像不是最新版的。
TensorFlow:1.6.04.====最近出了1.7,貌似仅支持cuda9
https://www.tensorflow.org/serving/serving_basic
首先第一步肯定是装好Ubuntu,其次装好对应显卡驱动,现在我们按照上面的顺序开始安装cuda。
cuda安装:
安装CUDA9.0?当时官网说的最新的TensorFlow支持cuda9,但是最新的已经到了9.1.所以我装9.0版
1 。首先去官网下载CUDA 9.0,就截了一个图,下载的是.run文件。
百度自己推荐的,点进去就有了。
https://developer.nvidia.com/cuda-90-download-archive?
target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
建议选择迅雷下载,360浏览器莫名其妙断了,下了一半的时候。迅雷也会断,但是你可以暂停一分钟,再去下快很多。也有人呢建议用火狐或者谷歌下,大家自己决定吧。
2. 找到Ubuntu你刚才下载文件里面安装
sudo sh cuda_9.0