ubuntu16.04配置cuda8.0+cudnn6+tensorflow1.4

安装相关依赖项

sudo add-apt-repository ppa:xorg-edgers/ppa
sudo apt-get update
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler 
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

安装nvidia驱动

首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合本身显卡的驱动:python

安装以前先卸载已经存在的驱动版本:
sudo apt-get remove --purge nvidia*
若电脑是集成显卡(NVIDIA独立显卡忽略此步骤),须要在安装以前禁止一项:
sudo service lightdm stop
执行如下指令安装驱动:
sudo apt-get install nvidia-384 #注意在这里指定本身的驱动版本!
安装完成以后输入如下指令进行验证:
sudo nvidia-smi
若列出了GPU的信息列表则表示驱动安装成功。
 
注:新主板的话UEFI,要把security boot关掉
重装显卡驱动:
关掉security boot
Ctrl+Alt+F1进入命令行模式
sudo service lightdm stop 关闭图形界面
sudo apt-get remove --purge nvidia* 删除原有驱动
sudo apt-get install nvidia-384 安装显卡驱动
sudo service lightdm start 开启图形界面
sudo reboot 重启
启动器/菜单等消失

重置compiz:linux

dconf reset -f /org/compiz/编程

重置compiz后,使用如下命令重启Unity:vim

setsid unityruby

此外,若是你想将Unity图标也进行重置,试试如下的命令吧:bash

unity --reset-icons网络

 

cuda安装

CUDA是NVIDIA的编程语言平台,想使用GPU就必需要使用cuda。
(1)下载CUDA
首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA:

下载完成后执行如下命令:
sudo sh cuda_8.0.27_linux.run
而后会出现一个神奇的界面,须要一直按enter键, 除了nvidia显卡选项必须选择no(这个咱们以前装过了)其余都按照默认来。
安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

而后设置环境变量和动态连接库,在命令行输入:app

$ sudo gedit /etc/profile

也可使用vim打开编程语言

sudo apt-get install vim-gtkpost

sudo vim /etc/profile

在打开的文件末尾加入:

export PATH=/usr/local/cuda/bin:$PATH

保存以后,建立连接文件:

sudo gedit /etc/ld.so.conf.d/cuda.conf

在打开的文件中添加以下语句:

/usr/local/cuda/lib64

而后执行

sudo ldconfig

使连接当即生效。
测试cuda的Samples:

cd /usr/local/cuda-7.5/samples/1_Utilities/deviceQuery make sudo ./deviceQuery

若是显示的是一些关于GPU的信息,则说明安装成功了。

cudnn6安装

cuDNN是GPU加速计算深层神经网络的库。

首先去官网(https://developer.nvidia.com/rdp/cudnn-download)下载cuDNN,可能须要注册一个帐号才能下载。

下载包:cudnn-8.0-linux-x64-v6.0.tgz 解压后获得 cuda 文件夹
在cuda 目录下执行以下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/

python+pip安装

sudo apt-get install python

For Python 3

sudo apt-get install python3-pip
vsudo apt-get install python3-pip python3-de

For Python 2

sudo apt-get install python-pip
sudo apt-get install idle2

sudo apt-get install idle3

安装tensorflow

去清华tensorflow找安装镜像
https://mirror.tuna.tsinghua.edu.cn/help/tensorflow/
选择本身匹配的tensorflow版本号就行了

sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/   https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.4.1-cp35-cp35m-linux_x86_64.whl

注:cudnn6支持的是tensorflow1.3及以上

相关文章
相关标签/搜索