Ubuntu 16.04 + GTX1080ti + CUDA8.0 + cudnn5.1 + tensorflow

来源:互联网 发布:svg js设置 transform 编辑:程序博客网 时间:2024/06/15 19:04

模块

  • 安装Ubuntu 16.04
  • 安装显卡驱动
  • 安装CUDA8.0
  • 安装cudnn5.1
  • 安装tensorflow
  • 安装Pycharm

安装Ubuntu 16.04

  • 分区:/boot:200M、swap:8G、/:>60G、/home:剩余全部,均为逻辑分区,起始位置,注意将引导项设置为/boot所在分区。
  • 分辨率调整,安装完毕后,由于没有安装显卡驱动,分辨率很低,可以手动修改一下grub文件:
    sudo vi /etc/default/grub
    添加:
    GRUB_GFXMODE=1920x1080

    sudo update-grub
    重启有效。

安装GTX1080ti驱动

sudo apt-get update
重启。
System Settings – Software&Updates – Additional Drivers – 选择Nvidia最新驱动(375)– Apply Changes.
重启。

安装CUDA8.0

下载cuda-xx_amd64.deb(网上都说deb方案安装无数坑,可是我并没有遇到什么坑)。
按照Nvidia官网安装说明检查:

$lspci | grep -i nvidia

如果没有显示正确的显卡型号:

$update-pciids

重新执行上一步。

$uname -m && cat /etc/*releasex86_64...
$gcc --version

Ubuntu 16.04自带的gcc版本应该是gcc5.4.0,但是CUDA8.0编译不支持gcc5.0版本以上,需要降级,降级方法:

sudo apt-get install g++-4.9sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30sudo update-alternatives --set cc /usr/bin/gccsudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30sudo update-alternatives --set c++ /usr/bin/g++

重新执行:

$gcc --version

应该会看到gcc版本降为4.9版本。

uname -r

cd到cuda安装包路径,如cd Download/

$md5sum <file>

检查完毕,如果检查无误,安装cuda:

$ sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb$ sudo apt-get update$ sudo apt-get install cuda

设置环境变量:
终端输入:

$ sudo gedit /etc/profile

文件末尾添加:

$ export PATH=/usr/local/cuda-8.0/bin:$PATH$ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64

使环境变量生效:

source ~/.bash_profile

重启。
检查CUDA版本:

$ nvcc –V

编译CUDA提供的例子:略

安装cudnn5.1

在官网下载最新的对应cuda8.0的cudnn5.1安装包,需要注册。
解压下载的cudnn5.1安装包,并将其放在/home目录下(自己可以选择)。

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

更新软连接:

cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.5sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5sudo ln -s libcudnn.so.5 libcudnn.so

更新链接库:

sudo ldconfig

完毕。

安装Anaconda2

安装tensorflow

见官网,推荐用Installing with virtualenv.

安装Pycharm

阅读全文
0 0