ubuntu,Nvidia驱动,cuda8+cudnn5.1详细安装过程

环境:

  • ubuntu 16.04
  • GTX 960

安装步骤

  • 安装Nvidia驱动

Nvidia驱动安装可以参考博客:https://blog.csdn.net/qq_20481015/article/details/82807709

  •  CUDA 8.0安装
sudo chmod +x cuda_8.0.61_375.26_linux.run
sudo sh cuda_8.0.61_375.26_linux.run --tmpdir=/tmp --override

根据命令行提示进行安装:

     按`q`退出条款浏览或者按`空格`直到条款末尾,输入 accept 接受条款;

     输入n不安装nvidia图像驱动,前面已经安装好驱动;

     输入y安装cuda 8.0 toolkit;

     回车确认cuda默认安装路径(/usr/local/cuda-8.0);

     输入y安装CUDA 8.0 Samples;

    输入CUDA 8.0 Samples安装路径:/home/用户名/CUDA/samples(选择你喜欢的目录进行 安装)。

配置cuda环境变量: 打开~/.bashrc文件

sudo gedit ~/.bashrc

     在bashrc文件的末尾追加下面内容:
 

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} 
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 

     使环境变量马上生效

source ~/.bashrc

然后,测试是否安装成功,输入:

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

如果显示:

./deviceQuery Starting...

 CUDA Device Query (Runtime API) version (CUDART static linking)

Detected 1 CUDA Capable device(s)

Device 0: "GeForce GT 740M"
  CUDA Driver Version / Runtime Version          8.0 / 8.0
  CUDA Capability Major/Minor version number:    3.5
  Total amount of global memory:                 2004 MBytes (2100953088 bytes)
  ( 2) Multiprocessors, (192) CUDA Cores/MP:     384 CUDA Cores
  GPU Max Clock rate:                            1032 MHz (1.03 GHz)
  Memory Clock rate:                             800 Mhz
  Memory Bus Width:                              64-bit
  L2 Cache Size:                                 524288 bytes
  Maximum Texture Dimension Size (x,y,z)         1D=(65536), 2D=(65536, 65536), 3D=(4096, 4096, 4096)
  Maximum Layered 1D Texture Size, (num) layers  1D=(16384), 2048 layers
  Maximum Layered 2D Texture Size, (num) layers  2D=(16384, 16384), 2048 layers
  Total amount of constant memory:               65536 bytes
  Total amount of shared memory per block:       49152 bytes
  Total number of registers available per block: 65536
  Warp size:                                     32
  Maximum number of threads per multiprocessor:  2048
  Maximum number of threads per block:           1024
  Max dimension size of a thread block (x,y,z): (1024, 1024, 64)
  Max dimension size of a grid size    (x,y,z): (2147483647, 65535, 65535)
  Maximum memory pitch:                          2147483647 bytes
  Texture alignment:                             512 bytes
  Concurrent copy and kernel execution:          Yes with 1 copy engine(s)
  Run time limit on kernels:                     No
  Integrated GPU sharing Host Memory:            No
  Support host page-locked memory mapping:       Yes
  Alignment requirement for Surfaces:            Yes
  Device has ECC support:                        Disabled
  Device supports Unified Addressing (UVA):      Yes
  Device PCI Domain ID / Bus ID / location ID:   0 / 1 / 0
  Compute Mode:
     < Default (multiple host threads can use ::cudaSetDevice() with device simultaneously) >

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0, CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GT 740M
Result = PASS

则表示安装成功:

  • cuDNN 5.1安装

首先,提取文件到当前文件夹

tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz 

然后,安装cudnn5.1.10 

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn* 
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5 
sudo ln -sf libcudnn.so.5 libcudnn.so 

注意,如果cudnn是其他版本,请对应修改,如安装cudnn5.1.5,应执行:

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn* 
sudo ln -sf libcudnn.so.5.1.5 libcudnn.so.5 
sudo ln -sf libcudnn.so.5 libcudnn.so 

猜你喜欢

转载自blog.csdn.net/qq_20481015/article/details/82807912