Linux下CUDA安装指南 (cuda install linux)
CUDA作为英伟达公司为自家GPU实现的通用计算平台,支持c++、VS、Python、MATLAB以及各种深度学习框架。为了方便使用CUDA,我们以下介绍,其中以Ubuntu系统为例。
1. 检查GPU型号
在Linux环境下,可以通过以下命令检查电脑上的GPU型号:
“`
lspci |grep -i nvidia
“`
其中-lspci是查看PCI设备信息,-grep是根据关键字查找相应信息。
2. 下载CUDA相关文件
CUDA的官方网站为https://developer.nvidi/cuda-toolkit-archive,用户可以根据自己的需要选择不同版本的CUDA Toolkit。
下载完成后,用户需要解压缩所下载的CUDA Toolkit包,建议将解压缩后的文件夹放在/opt目录下,方便管理。
3. 安装依赖
需要安装的依赖包括:gcc、g++、make、ssh server、libc6、libc-dev-bin、libc-dev、libcublas6.5、libncurses5、libncurses6、zlib1g、zlib1g-dev、libjpeg-dev、nvidia驱动等。
“`
sudo apt-get install -y ssh gcc g++ libc6-dev libcublas10 libcublas-dev zlib1g zlib1g-dev libjpeg-dev
sudo apt-get install -y linux-headers-$(uname -r)
“`
其中uname -r可以获取当前操作系统内核版本及其指示。
4. 编辑环境变量
CUDA需要在系统环境变量中添加一些变量,以便正常使用。用户需要在/etc/environment中添加以下环境变量:
“`
PATH=”$PATH:/usr/local/cuda/bin”
LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64″
“`
其中/usr/local/cuda为CUDA的解压目录。
5. 修改系统配置文件
在Ubuntu系统中,由于默认启用了nouveau开源驱动,因此需要先将其禁用,才能使用nvidia官方驱动。
“`
sudo gedit /etc/modprobe.d/blacklist.conf
“`
在文件内容的末尾加入以下内容:
“`
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
“`
保存后执行以下命令:
“`
sudo update-initramfs -u
sudo reboot
“`
重启后,用户可以安装NVIDIA官方显卡驱动:
“`
sudo apt-get install nvidia-driver-*
“`
6. 进行CUDA安装
用户可以通过以下命令进入到解压后的文件夹当中:
“`
cd /opt/cuda-10.1/
“`
可以使用以下命令安装CUDA:
“`
sudo sh cuda_10.1.168_418.67_linux.run //其中cuda_10.1.168_418.67_linux.run为相应的文件名
“`
安装过程中会弹出一些选项供用户选择(包含PATH、nvcc、driver的安装路径、CUDA示例、NVIDIA Accelerated Graphics Driver for Linux-x86_64等),请根据自己的需要进行选择。
7. 测试CUDA是否安装成功
安装完成后,用户可以使用以下命令查看CUDA的版本号:
“`
nvcc –version
“`
如果输出相应版本信息,则说明CUDA安装成功。
8. 安装cuDNN和TensorRT
cuDNN是deep learning体系上的核心库,而TensorRT是针对深度学习优化的高性能的推理引擎,是英伟达的一项高性能机器学习软件开发工具包。用户可以在官方网站下载相应的安装包,然后使用以下命令进行安装:
“`
tar –xzvf .tgz //解压安装包
sudo cp –r cuda/include/* /usr/local/cuda-10.1/include/ //将解压后的文件拷贝到相应目录下
sudo cp –r cuda/lib64/* /usr/local/cuda-10.1/lib64/
“`
安装完成后,用户可以使用cuDNN的自带代码进行测试,以确定安装是否成功。
本文针对Linux下CUDA的安装过程进行了详细的介绍,以Ubuntu 18.04 LTS操作系统为例,向读者提供了一份完整、实用的CUDA安装指南。需要注意的是,在安装时需要结合当前电脑的系统情况和自己的应用场景进行选择,以确保安装成功。