Linux

centos 7 安装CUDA9.0 +CUDNN

2018-05-01  本文已影响0人  Persistently

先根据系统实际情况下载对应的CUDA,这里我下载了CUDA 9.0 版本

1.检查是否安装了GPU:

   lspci | grep -i nvidia  

2.安装gcc、g++编译器

   sudo yum install gcc    
   sudo yum install gcc-c++  

3.安装kernel-devel

 sudo yum install kernel-devel  

4.安装Driver,Toolkit和Samples

  sudo sh cuda_9.0.176_384.81_linux-run.run

在这里我们是自己安装了匹配的驱动,所以第一项Driver出来的时候选择N 后面全是Y ,即可

5.在用户根目录下的.bashrc文件中加入如下内容:

    export PATH=$PATH:/usr/local/cuda-9.0/bin    
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64  
   保存退出
    source .bashrc (立即生效文件)
   如果是全环境变量(所用用户都能使用),需要在/etc/profile文件加入上面的几句话

CUDNN

  1. 选择合适的版本(与CUDA版本对应)


    CUDNN

    2.解压CUDNN

    tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz
    
  2. 复制相关文件到cuda特定目录下(我的cuda安装目录为/usr/local/cuda-9.0/)

    sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
    
  3. 修改文件权限

    sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
    

  1. centos 7 下安装 python 的 pip

    yum update


上一篇 下一篇

猜你喜欢

热点阅读