Ubuntu14.04+cuda7.5+cudnn-v5+tensorflow-0.10(GPU加速)安装
先下载NVIDIA公司的cuda-7.5的安装说明,按照说明里面的预安装步骤走一遍,安装的时候,不要直接使用NVIDIA公司给的deb和run文件,都会出现安装不成功现象。
1、如果安装不成功,先运行下面两条指令卸载驱动和工具:
sudo /usr/local/cuda-7.5/bin/uninstall_cuda_7.5.pl
sudo /usr/bin/nvidia-uninstall/
2、接着先安装显卡对应版本英伟达驱动(具体在官网查询)
驱动官网:http://www.nvidia.com/Download/index.aspx?lang=en-us
查询完之后不需要下载官网的驱动,试过 也是坑
接着进行下面步骤
添加源
sudo add-apt-repository ppa:graphics-drivers/ppa
更新
sudo apt-get update
安装
sudo apt-get install nvidia-375
重启
sudo reboot
3、官方下载cuda7.5.run版本
cd 到对应目录下进行安装,安装命令
sudo sh ./cuda_8.0.44_linux.run
先空格到最后, 第二个安装驱动选择n,其他选y和回车,安装好之后重启,进入图形界面,执行
sudo nano ~/.bashrc
在文件末尾添加
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
运行nvidia-smi,显示NVIDIA-SMI 375.20等
则cuda安装成功。还可以运行几个例子,输入如下代码
cd到样本文件夹下,选择1_Utilities/deviceQuery
make 如果报错没有g++,安装,命令如下
sudo apt-get install g++
下面执行
sudo ./deviceQuery
如果不能执行,输入sudo su 在root下执行,结果最后PASS
3、安装cudnn-7.5-v5.1,在英伟达官网下载,这个是和cuda7.5对应的,接着运行
解压
cd /tmp
tar -zxf cudnn-7.5-linux-x64-v5.1.tgz
cd ./cuda
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h/usr/local/cuda/include/
接着进入文件夹/usr/local/cuda/lib64/,并查看cudnn软连接是否存在,存在删除
cd /usr/local/cuda/lib64/
ls -al
rm -rflibcudnn.so libcudnn.so.5
接着重新创建软连接
sudo ln -sf libcudnn.so.5.1.3libcudnn.so.5
sudo ln -sflibcudnn.so.5 libcudnn.so
设置环境变量,将/usr/local/cuda/lib64加入cuda.conf
sudo nano /etc/ld.so.conf.d/cuda.conf
使得设置有效
sudo ldconfig
更改cuda使得支持高于5.0版本的gcc,编辑/usr/local/cuda/include/host_config.h第115行
sudo nano /sur/local/cuda/include/host_config.h
改变:
change
#error --unsupported GNU version! gcc versions later than 4.9 are not supported!
to
//#error --unsupported GNU version! gcc versions later than 4.9 are not supp orted!
运行样本,进入/usr/local/cuda/cuda_samples/NVIDIA_CUDA-7.5_Samples运行
sudo make all-j8 (8代表8核CPU)
接着进入/bin/x86_64/linux/release运行
./deviceQuery
得到PASS,成功
3、使用pip安装TensorFlow
sudo apt-getinstall python-pip python-dev
github下载GPU版本tensorflow-0.10.0-cp27-none-linux_x86_64.whl,运行
$ sudo apt-getinstall python-pip python-dev
并到对应放置tensorflow-0.10.0-cp27-none-linux_x86_64.whl文件夹下,运行
$ sudo pipinstall --upgrade tensorflow-0.10.0-cp27-none-linux_x86_64.whl
如果第一次失败,再输入
$ sudo pipinstall --upgrade tensorflow-0.10.0-cp27-none-linux_x86_64.whl
成功