ubuntu 如何安装cuda

在Ubuntu系统上安装CUDA（Compute Unified Device Architecture）需要几个步骤，因为CUDA主要用于支持NVIDIA GPU的深度学习和科学计算任务。以下是基本的安装流程：

1. **检查GPU兼容性**：
首先确保你的GPU是NVIDIA并且受CUDA支持。访问NVIDIA官网（https://developer.nvidia.com/cuda-gpus）确认兼容性。

2. **添加NVIDIA repository**：
更新系统包列表，并添加NVIDIA的apt源：

   sudo apt-get update
   sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/repos/ubuntu/xenial/local.gpg
   echo "deb [arch=amd64] https://developer.download.nvidia.com/repos/ubuntu/xenial upgradable non-free" > /etc/apt/sources.list.d/nvidia-xconfig.list

3. **安装CUDA**：
更新源并安装CUDA：

   sudo apt-get upgrade
   sudo apt-get install cuda

这可能需要多次输入密码，并且可能需要下载较大的文件。

4. **配置环境变量**：
完成安装后，你需要设置环境变量让系统识别CUDA。这通常涉及到编辑`/etc/bash.bashrc`文件，添加如下内容：

   export PATH=/usr/local/cuda/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

5. **验证安装**：
最后，你可以通过命令行测试CUDA是否安装成功，例如运行`nvcc --version`。

注意：安装过程可能会因硬件型号和Ubuntu版本的不同而略有差异。如果你遇到任何问题，建议查阅NVIDIA的官方文档或在线教程。