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)查看适合自己显卡的驱动:

安装之前先卸载已经存在的驱动版本:
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:

dconf reset -f /org/compiz/

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

setsid unity

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

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}}

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

$ sudo gedit /etc/profile

也可以使用vim打开

sudo apt-get install vim-gtk

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
sudo apt-get install python3-pip python3-dev

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及以上

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

posted on 2017-12-12 21:20  J_K_Guo  阅读(4175)  评论(0编辑  收藏  举报