Cuda9.0安装

CUDA 9.0安装笔记

最近实验室新购买两块K80的GPU。作为好奇的小猪,当然会自报奋勇去配置环境。在这篇博客中将会介绍在centos7下配置CUDA 9.0的步骤。

1. 什么是CUDA?

引用官方介绍

CUDA® is a parallel computing platform and programming model invented by NVIDIA. It enables dramatic increases in computing performance by harnessing the power of the graphics processing unit (GPU).

通过这句话可以了解到,CUDA是一个由NVIDIA公司发明的并行计算平台和编程模型。它利用图形处理单元(GPU)的强大功能,使计算性能大幅提高。

cuda主要具有以下两个方面的作用:

  1. 提供向C这样的标准语言的扩展,从而实现并行算法的简单实现。通过使用CUDA C/C++可以是编程人员专注于算法的并行化而不是程序本身的实现
  2. 支持应用程序同时使用CPU和GPU进行异构计算。应用程序的串行部分运行在CPU上,并行部分则运行在GPU上。

总之,CUDA为我们提供了操作GPU的high level接口,通过调用这接口可以简化程序的编写过程。

2. 安装环境介绍

项目 -
操作系统 CentOS Linux release 7.4.1708 (Core)
内核版本 3.10.0-693.5.2.el7.x86_64
内存 128GB
硬盘 6TB
运算卡 K80

3. 安装CUDA 9.0

安装CUDA 9.0分为两步:1、安装依赖环境 2、安装CUDA 9.0。

3.1. 安装依赖环境

通过lspci | grep -i nvidia查看机器是否已经识别出GPU卡。如果没有显示GPU卡,则需要运行update-pciids更新一下。

lspci

通过运行gcc --version检测本机是否已经安装了gcc 4.8.5。如果没有安装gcc 4.8.5则通过yum install gcc gcc-c++进行安装。

安装kernel headers和相关开发包
yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)

好,到现在为止依赖环境已经安装的差不多了。

3.2. 安装CUDA 9.0

CUDA具有两种安装方式:

  1. distribution-specific packages方式,可以和系统的原生库管理系统对接
  2. distribution-independent package方式,适应更多的Linux系统,不过不更新系统的原生库管理系统

本猪猪比较好奇,所以两种方法都进行了测试,所以接下来将会分别介绍两种安装方式。

如果你的机器以前安装过CUDA,则需要将老版本的CUDA卸载掉,下面是卸载方式。因为本猪猪用的是崭新的机器,所以就先跳过这一步啦。

# 1. 通过runfile方式安装的
# 卸载cuda toolkit
sudo /usr/local/cuda-X.Y/bin/uninstall_cuda_X.Y.pl
# 卸载NVIDIA 驱动
sudo /usr/bin/nvidia-uninstall

# 2. 通过RPM/Deb方式安装的
sudo yum remove cuda* 	#Redhat/CentOS
sudo apt-get --purge remove cuda* 	#Ubuntu
3.2.1. distribution-specific packages方式

1、安装DKMS依赖:

fedoraproject上面下载DMKS安装包,Centos7 对应的安装包路径为http://dl.fedoraproject.org/pub/epel/7/x86_64/Packages/d/dkms-2.4.0-1.20170926git959bd74.el7.noarch.rpm

安装DKMS依赖yum install -y elfutils-libelf-devel

安装DKMS包rpm -ivh dkms-2.4.0-1.20170926git959bd74.el7.noarch.rpm

2、安装元数据包源:

下载地址:cuda-repo-rhel7-9.0.176-1.x86_64.rpm)

安装:sudo rpm -ivh cuda-repo-rhel7-9.0.176-1.x86_64.rpm

3、更新元数据

sudo yum clean expire-cache
sudo yum update

4、安装CUDA

直接运行一下命令进行安装,当然如果你不放心的话,可以先执行sudo yum info cuda查看CUDA包的相关信息。

sudo yum install cuda
sudo yum update cuda-drivers

好啦,现在已经安装成功啦!

运行nvidia-smi查看显卡信息,试试?

nvidia-smi

运行cat /usr/local/cuda/version.txt查看CUDA版本,试试?

CUDA-version

3.2.2. distribution-independent package方式

抱歉了各位,本猪猪有些累了,需要休息系。稍后再来补充。

参考文档

  1. cuda-installation-guide-linux
posted @ 2017-11-26 09:18  WingPig99  阅读(8527)  评论(0编辑  收藏  举报