模块
- 安装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=1920x1080sudo 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/*release
x86_64
...
$gcc --version
Ubuntu 16.04自带的gcc版本应该是gcc5.4.0,但是CUDA8.0编译不支持gcc5.0版本以上,需要降级,降级方法:
sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo 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.5
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
更新链接库:
sudo ldconfig
完毕。
安装Anaconda2
略
安装tensorflow
见官网,推荐用Installing with virtualenv.
安装Pycharm
略