淘先锋技术网

首页 1 2 3 4 5 6 7

一,安装显卡驱动
在 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