一,安装显卡驱动
在 Ubuntu 系统中的 System Settings -> Software & Updates -> Additional Drivers 安装相应的显卡和CPU驱动,重启后在 System Settings -> Details 中可以看到自己的显卡型号则说明驱动已经安装成功,此时在命令行下输入 nvidia-smi 也可以看到显卡的相关信息。
如果Additional Drivers没有显卡驱动的信息的话,使用下面的命令把显卡驱动加入PPA中:
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
二,安装cuda
1.到 CUDA 官网 下载合适版本的 runfile 文件(强烈推荐), 然后在终端运行 sudo sh cuda_filename.run,除了在询问你是否要安装驱动的时候选择 No(上面已安装过),其他都可以选择 Yes。
2.在家目录打开终端运行 sudo gedit .bashrc
,在文件末尾添加相应的路径。
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PA
3.在终端运行source .bashrc
4.终端运行 nvcc -V
,出现相关 CUDA 版本信息说明安装。
三, 安装 CUDNN
1.解压对应版本的 CUDNN 压缩包,复制相应文件到相关路径。
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
查看cudnn版本的命令
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2