CUDA为NVIDIA提供的运算平台，包含了CUDA指令集架构以及GPU内部的并行计算引擎。若想要在NVIDIA系列GPU上运行GPU加速计算任务，需要安装CUDA环境。

CUDA安装

安装前准备

步骤1、确认GPU驱动已安装。

nvidia-smi

步骤2、确认适配的CUDA版本。

安装步骤

步骤1、拿到适配的CUDA版本，在CUDA Toolkit Archive查找适配CUDA软件包。

步骤2、点击进入，找到适配的runfile。

步骤3、执行下载命令，注意检查CUDA版本和GPU驱动版本是否一致。

wget https://developer.download.nvidia.com/compute/cuda/<CUDA_VERSION>/local_installers/cuda_<CUDA_VERSION>_<DRIVER_VERSION>_linux.run

步骤4、执行安装命令。

wget https://developer.download.nvidia.com/compute/cuda/12.9.1/local_installers/cuda_12.9.1_575.57.08_linux.run

若已安装GPU驱动需要取消勾选Driver，若未安装GPU驱动请勾选安装。

步骤5、执行命令配置CUDA环境变量。

echo 'export PATH=/usr/local/cuda/bin:$PATH' | sudo tee /etc/profile.d/cuda.sh
source /etc/profile

步骤6、检查安装是否成功。

echo 'export PATH=/usr/local/cuda/bin:$PATH' | sudo tee /etc/profile.d/cuda.sh
source /etc/profile