centos 7 安装CUDA9.0 +CUDNN

先根据系统实际情况下载对应的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


©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容