Ubuntu 深度炼丹环境配置
深度炼丹最麻烦的就是环境配置了,下面过程记录了本人进行深度炼丹环境的配置。
首先是安装图形显卡驱动,按如下指令进行即可
sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get updatesudo apt-get upgradesudo 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.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/include/sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -dsudo chmod a+r /usr/local/cuda/include/cudnn.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*