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主要具有以下兩個方面的作用:
- 提供向C這樣的標准語言的擴展,從而實現並行算法的簡單實現。通過使用CUDA C/C++可以是編程人員專注於算法的並行化而不是程序本身的實現
- 支持應用程序同時使用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
更新一下。
通過運行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具有兩種安裝方式:
- distribution-specific packages方式,可以和系統的原生庫管理系統對接
- 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
查看顯卡信息,試試?
運行cat /usr/local/cuda/version.txt
查看CUDA版本,試試?
3.2.2. distribution-independent package方式
抱歉了各位,本豬豬有些累了,需要休息系。稍后再來補充。