Ubuntu安装CUDA

安装CUDA前，请先确保您的系统满足以下要求：

- NVIDIA显卡
- Ubuntu操作系统
- 安装了编译器和开发工具（如gcc、g++、make等）
- 最新的NVIDIA驱动程序

以下是在Ubuntu上安装CUDA的步骤：

1. 下载CUDA Toolkit

访问NVIDIA官网下载CUDA Toolkit。选择您要安装的CUDA版本和Ubuntu版本。例如，对于CUDA 11.2和Ubuntu 20.04，您可以通过以下命令下载CUDA Toolkit：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

2. 安装CUDA Toolkit

运行以下命令安装CUDA Toolkit：

sudo apt-get install cuda

3. 配置环境变量

安装完成后，需要配置环境变量。将以下内容添加到`~/.bashrc`文件中：

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

保存并关闭文件。运行以下命令使环境变量立即生效：

source ~/.bashrc

4. 验证安装

运行以下命令验证CUDA是否正确安装：

nvcc -V

如果输出类似于以下内容，则表示CUDA已成功安装：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Tue_Sep_15_19:09:53_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29069683_0