Ubuntu 深度炼丹环境配置

Ubuntu 深度炼丹环境配置

深度炼丹最麻烦的就是环境配置了，下面过程记录了本人进行深度炼丹环境的配置。

首先是安装图形显卡驱动，按如下指令进行即可

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install nvidia-396

这样便可以安装完图形显卡驱动。

紧接着就是安装CUDA，这是一个比较麻烦的工作。首先是从官网上下载最新的CUDA版本，下面的命令也可以，但是安装得到的版本通常不是最新的。

sudo apt install nvidia-cuda-toolkit

下载runfile即可，安装过程较为简单。为了验证安装是否成功，可以按如下步骤进行测试：

查看cuda 的版本号，输入如下指令即可：

nvcc -V

编译CUDA samples。找到你安装CUDA的过程中，选择的samples保存位置，进入后运行make命令，然后可以在Samples里面找到bin/x86_64/linux/release/目录，并切换到该目录  运行deviceQuery程序，sudo ./deviceQuery pass即通过。

最后一步最烦人，目前亲自试过的比较正确的方式是如下

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*