Linux下安装cuda
### Linux下安装CUDA详解 #### 一、简介 CUDA (Compute Unified Device Architecture) 是由NVIDIA公司开发的一种并行计算平台和技术,它允许开发者利用NVIDIA GPU的强大计算能力来加速计算密集型应用。本文将详细介绍如何在Linux环境下安装CUDA,并解决安装过程中可能遇到的一些常见问题。 #### 二、禁用nouveau驱动 为了确保CUDA能够正常工作,首先需要禁用Linux系统自带的nouveau图形驱动。这是因为nouveau驱动与官方NVIDIA驱动存在冲突。 1. **禁用nouveau**: - 打开终端。 - 运行命令 `sudo vim /etc/modprobe.d/blacklist.conf`。 - 在文件中添加以下内容: ```plaintext blacklist vga16fb blacklist rivafb blacklist rivatv blacklist nvidiafb options nouveau modeset=0 ``` - 保存文件后,运行 `sudo update-initramfs -u` 更新配置。 - 检查nouveau是否已被禁用: `$ lsmod | grep nouveau`。如果没有输出,则表示已成功禁用。 #### 三、关闭UEFI安全启动 部分现代计算机采用UEFI安全启动模式,这可能会阻止NVIDIA驱动程序的安装。因此,在安装CUDA之前,需要关闭BIOS中的UEFI安全启动选项。 - 进入计算机BIOS设置界面。 - 寻找与“UEFI”或“Secure Boot”相关的选项,并将其设置为禁用状态。 - 保存更改并重启计算机。 #### 四、安装必要的依赖包 在安装CUDA之前,需要确保系统已经安装了一些基本的开发工具和库。 1. **安装基本开发工具**: - 运行命令 `sudo apt-get install build-essential pkg-config linux-headers-$(uname -r)`。 2. **安装额外库**: - 如果前面的命令安装失败,可以尝试安装以下库: - `sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5` - `sudo apt-get install --no-install-recommends libboost-all-dev` - `sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev` - `sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev` #### 五、下载并安装CUDA 1. **下载CUDA安装包**: - 访问NVIDIA官网 [https://developer.nvidia.com/cuda-downloads](https://developer.nvidia.com/cuda-downloads) 下载最新版本的CUDA安装包。 - 笔者选择下载了runfile格式的安装包。 2. **安装CUDA**: - 运行命令 `sudo sh <下载安装包名>.run` 开始安装。 - 在安装过程中,接受许可协议,并根据提示进行默认设置安装。 - 安装完成后,还需要配置环境变量。 3. **配置环境变量**: - 添加以下内容到 `.bashrc` 文件中: ```bash echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc ``` #### 六、安装NVIDIA驱动 1. **通过PPA安装NVIDIA驱动**: - 添加PPA仓库:`sudo add-apt-repository ppa:graphics-drivers/ppa` - 更新软件列表:`sudo apt-get update` - 安装NVIDIA驱动:`sudo apt-get install nvidia-352` - 重启计算机。 #### 七、验证安装 1. **检查NVIDIA驱动版本**: - 运行命令 `cat /proc/driver/nvidia/version`。 2. **检查CUDA版本**: - 运行命令 `nvcc -V`。 #### 八、解决常见问题 1. **问题:CUDAdriver version is insufficient for CUDAruntime version** - 解决方案:升级NVIDIA驱动至兼容版本,如 `sudo apt install nvidia-375`。 2. **问题:no CUDA-capable device is detected** - 解决方案:确保NVIDIA驱动已正确安装,并重启计算机。 #### 结语 安装CUDA的过程中可能会遇到各种问题,但只要耐心解决,最终都能够顺利完成。希望每位读者都能顺利安装并享受CUDA带来的性能提升。在遇到困难时,请不要放弃,寻求帮助并继续前进!