centos中cuda驱动及其相关包cudnn安装

来源:互联网 发布:搜狐网络大厦一层图片 编辑:程序博客网 时间:2024/06/14 13:22

有很多没有使用过GPU的同学总认为在使用GPU之前要安装两个驱动,一个是nvidia驱动,另一个是cuda驱动,但这种想法是错误的,只需安装一个cuda驱动就可以了。

1、lspci | grep -i nvidia


2、uname -m && cat /etc/*release


3、gcc --version


如果没有检测出则:yum install gcc gcc-c++

4、uname -r


5、sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)

此处可以把编码去掉,直接安装yum install kernel-devel.x86_64,但去掉之后安装的是最新版本的核,因此不可取,即使安装成功后也会出现nvidia-smi检测不到驱动的后果

最终的解决办法是在网上下载上一部得出的文件名字,重新安装一遍:sudo rpm -i kernel-devel-3.10.0-514.e17.x86_64

6、sudo sh cuda_8.0.61_375.26_linux.run

7、修改环境变量

临时:

export CUDA_HOME=/usr/local/cuda-8.0

export PATH= /usr/local/cuda-8.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib:${LD_LIBRARY_PATH}"


永久:vim ~/.bashrc

加入如下几行

  1. export CUDA_HOME=/usr/local/cuda-8.0  
  2. export PATH=/usr/local/cuda-8.0/bin:$PATH  
  3. export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH  
  4. export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib:${LD_LIBRARY_PATH}"  

8、测试 此图是截取别人的


9、 tar -xvzf cudnn-8.0-linux-x64-v6.0.tgz

cp-P cuda/include/cudnn.h /usr/local/cuda-8.0/include

cp-P cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64

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


准备安装包

  • NVIDIA-Linux-x86_64-381.22.run # 最新显卡驱动(没用)
  • cuda_8.0.61_375.26_linux.run # 最新CUDA安装包(安装)
  • cudnn-8.0-linux-x64-v6.0.tgz # cudnn库v6.0(没用)
  • cudnn-8.0-linux-x64-v5.1.tgz # cudnn库v5.1 (用到)

参考:http://blog.csdn.net/itaacy/article/details/72628792?utm_source=itdadao&utm_medium=referral

http://blog.csdn.net/Yaphat/article/details/53648818

http://blog.csdn.net/u012325865/article/details/73034018

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=rpmlocal


原创粉丝点击