深度学习环境搭建:Tensorflow1.4.0+Ubuntu16.04+Python3.5+Cuda8.0+Cudnn6.0
Reference
硬件说明:
显卡:gtx-1080,8G显存
NVIDIA Corporation Device 1b80
软件准备:
深度学习环境必须适配,不能版本之间必须匹配,这些库之间有依赖关系。
我的环境用的是:
ubuntu 16.04
Python3.5 # tensorflow无法完全支持3.6版本的,3.5比较稳定
Cuda8.0 # 目前Ubuntu16.04对cuda8.0的版本支持较好,同时Tensorflow对cuda9.0不太友好,因此我们选择Cuda8.0
Cudnn6.0
Tensorflow1.4.0
1. 安装Ubuntu16.04
从Ubuntu官网下载镜像,并用U盘安装。
安装完成之后更新系统。
sudo apt-get update
sudo apt-get upgrade
2. 安装显卡驱动
亲测可用,但是需要注意需要去英伟达官网下载适合自己电脑的版本(nvidia网页可以自己测出你的电脑所需要的型号),具体安装过程参考以上博客。
重启电脑,通过nvidia-smi命令查看驱动信息,如果成功显示,那么驱动安装成功
3.安装Cuda8.0
我们选择linux平台,下载runfile。执