您好,欢迎访问木子天禾科技官网!
17319297980
15934152105
北京软件开发公司_山西软件开发公司_木子天禾科技官网
联系我们

北京软件开发公司_山西软件开发公司_木子天禾科技官网

地址:山西省太原市小店区恒大未来城5-1711
手机:15934152105

咨询热线17319297980

centos 7 安装CUDA9.0 +CUDNN

发布时间:2021-01-09 20:41人气:

CUDA

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

1.检查是否安装了GPU:


  1.  
    lspci | grep -i nvidia

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


  1.  
    sudo yum install gcc
  2.  
    sudo yum install gcc-c++

3.安装kernel-devel


  1.  
    sudo yum install kernel-devel

4.安装Driver,Toolkit和Samples


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

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

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


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

推荐资讯

17319297980