我们在安装CUDA的时候要留意不同的ubuntu系统对应不同的版本;同时需要注意我们的ubuntu系统要选用64bit的,我在使用32bit的桌面ubuntu时,出现了CUDA无法正常使用的问题。
为了使用CUDA,我们首先得安装nVIDIA Driver驱动程式,然后安装CUDA Toolkit工具;最后我们可以通过CUDA Samples来检测CUDA是否可以正常工作。
具体的安装步骤:
A.安装nVIDIA Driver驱动
如果使用 cuda_xxxx.run 來安裝 nVIDIA driver,会出现Ubuntu 桌面无法开启的情况,所以我们使用以下的步骤来安裝 nVIDIA driver。
1.首先安装开发必须套件:
sudo apt-get install build-essential g++ vim
2.接着安装 nVIDIA显示卡 driver,安装之前必须将 Ubuntu 的 x window关掉。切换到文字界面:
Ctrl+Alt+F1
3.关掉x window界面:
sudo service lightdm stop
4.添加驱动的下载源:
sudo add-apt-repository ppa:xorg-edgers/ppa
sudo apt-get update
5.安装340版本的驱动程序:
sudo apt-get install nvidia-340
6.安裝 CUDA UVM Module
sudo apt-get install nvidia-340-uvm
7.重新开机
B.安装CUDA Toolkit工具
1.首先下载对应版本的CUDA,这里我们以run方式为例:
https://developer.nvidia.com/cuda-downloads
2.CUDA下载完成之后,到对应的文件夹下,赋予cuda_xxxx.run可执行的权限;这里假设CUDA下载到/home/user/download下:
cd /home/user/download/
chmod 755 ./cuda_xxxx.run
3.将刚刚下载的cuda_xxxx.run解压为三个文件:
cuda-xxxx.run(CUDA Toolkit)
NVIDIA-xxxx.run(nVIDIA Driver)
cuda-samplesxxxx.run(CUDA Samples)
cd /home/user/
mkdir ./temp
./cuda_xxxx.run --extract=/home/user/temp
cd /home/user/temp
4.安装CUDA Toolkit
sudo ./cuda-xxxx.run
5.安装好CUDA Toolkit之后,设定系统环境变量路径;修改.bashrc
vim ~/.bashrc
在档案最后两行加上:
export PATH=/usr/local/cuda-xxx/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xxx/lib64:$LD_LIBRARY_PATH
6.重载环境变量
source ~/.bashrc
C.安装CUDA Samples进行CUDA是否安装成功的检测
1.安装CUDA Samples
sudo ./cuda-xxxx.run
2.完成安装之后,编译sample,检测CUDA是否正常运行:
sudo su
cd /usr/local/cuda-xxx/sample/1_Utilities/deviceQuery
make
./deviceQuery